Merge pull request #302 from JonnyWong16/feature/hide_episode_spoilers

Add option for summary prefix in hide_episode_spoilers.py
This commit is contained in:
blacktwin 2021-08-26 23:12:48 -04:00 committed by GitHub
commit 57afb59f0b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -19,6 +19,8 @@
# --rating_key {rating_key} --image spoilers.png
# To blur the episode artwork (optional blur in pixels):
# --rating_key {rating_key} --blur 25
# To add a prefix to the summary (optional string prefix):
# --rating_key {rating_key} --summary_prefix "** SPOILERS **"
# * Watched (optional):
# --rating_key {rating_key} --remove
@ -66,6 +68,7 @@ if __name__ == "__main__":
parser.add_argument('--rating_key', required=True, type=int)
parser.add_argument('--image')
parser.add_argument('--blur', type=int, default=25)
parser.add_argument('--summary_prefix', nargs='?', const='** SPOILERS **')
parser.add_argument('--remove', action='store_true')
opts = parser.parse_args()
@ -99,7 +102,11 @@ if __name__ == "__main__":
# Delete the episode artwork image file
os.remove(os.path.join(episode_folder, filename))
elif opts.image:
# Unlock the summary so it will get updated on refresh
episode.edit(**{'summary.locked': 0})
continue
if opts.image:
# File path to episode artwork using the same episode file name
episode_artwork = os.path.splitext(episode_filepath)[0] + os.path.splitext(opts.image)[1]
# Copy the image to the episode artwork
@ -115,5 +122,12 @@ if __name__ == "__main__":
with open(episode_artwork, 'wb') as f:
shutil.copyfileobj(blurred_artwork, f)
if opts.summary_prefix and not episode.summary.startswith(opts.summary_prefix):
# Use a zero-width space (\u200b) for blank lines
episode.edit(**{
'summary.value': opts.summary_prefix + '\n\u200b\n' + episode.summary,
'summary.locked': 1
})
# Refresh metadata for the episode
episode.refresh()