diff --git a/stream/stream.py b/stream/stream.py index 541a60a..0dd3a71 100644 --- a/stream/stream.py +++ b/stream/stream.py @@ -7,20 +7,20 @@ import time from datetime import datetime def getCreationTime(img): - creationtime = time.ctime(path.getctime(img)) - date = datetime(creationtime).strftime('%y%m%d') - print('creationtime', date) - return date + creationtime = time.ctime(path.getctime(img)) + date = datetime(creationtime).strftime('%y%m%d') + print('creationtime', date) + return date def getStream(generator): - for page in generator.pages: - dir = "/var/www/stream" - if path.exists(dir): - images = [(getCreationTime(dir+'/'+f), f) for f in listdir(dir) if path.isfile(path.join(dir, f))] - else: # if the website is generated locally - images = ['placeholder.png'] - print('>>> The stream is streaming!', images) - page.stream = images + for page in generator.pages: + dir = "/var/www/stream" + if path.exists(dir): + images = [(getCreationTime(dir+'/'+f), f) for f in listdir(dir) if path.isfile(path.join(dir, f))] + else: # if the website is generated locally + images = ['placeholder.png'] + print('>>> The stream is streaming!', images) + page.stream = images def register(): - signals.page_generator_finalized.connect(getStream) + signals.page_generator_finalized.connect(getStream)