iou1name/fileHost
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

changed scheduling to apscheduler

Browse Source
This commit is contained in:
iou1name 2018-06-12 08:51:54 -04:00
parent 671e7a107d
commit a73641de0d
1 changed files with 18 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

40
fileHost.py
View File

@ -17,6 +17,7 @@ from flask import Flask, session, request, abort, redirect, url_for, g, \
render_template render_template
from werkzeug.utils import secure_filename from werkzeug.utils import secure_filename
from flask_paranoid import Paranoid from flask_paranoid import Paranoid
from apscheduler.schedulers.background import BackgroundScheduler
class ReverseProxied(object): class ReverseProxied(object):
""" """
@ -53,22 +54,6 @@ class ReverseProxied(object):
return self.app(environ, start_response) return self.app(environ, start_response)
class CronThread(threading.Thread):
def __init__(self):
threading.Thread.__init__(self)
self.stop = threading.Event()
def run(self):
while not self.stop.is_set():
records = db_execute(
"SELECT filename, delete_date FROM uploads WHERE delete_date"
).fetchall()
for filename, delete_date in records:
if time.time() >= delete_date:
delete_file(filename)
self.stop.wait(60)
app = Flask(__name__) app = Flask(__name__)
app.wsgi_app = ReverseProxied(app.wsgi_app) app.wsgi_app = ReverseProxied(app.wsgi_app)
app.config['MAX_CONTENT_LENGTH'] = 128 * 1024 * 1024 app.config['MAX_CONTENT_LENGTH'] = 128 * 1024 * 1024
@ -80,6 +65,9 @@ app.config["DB_LOCK"] = threading.Lock()
paranoid = Paranoid(app) paranoid = Paranoid(app)
paranoid.redirect_view = 'login' paranoid.redirect_view = 'login'
scheduler = BackgroundScheduler(timezone="America/New_York")
scheduler.start()
def db_execute(*args, **kwargs): def db_execute(*args, **kwargs):
""" """
@ -94,6 +82,19 @@ def db_execute(*args, **kwargs):
return res return res
@scheduler.scheduled_job("interval", minutes=1)
def delete_this():
"""
Removes files that are past the expiration date.
"""
records = db_execute(
"SELECT filename, delete_date FROM uploads WHERE delete_date"
).fetchall()
for filename, delete_date in records:
if time.time() >= delete_date:
delete_file(filename)
def init(): def init():
""" """
Initializes the application. Initializes the application.
@ -127,11 +128,6 @@ def init():
"uploaded_date INTEGER DEFAULT (STRFTIME('%s', 'now'))," "uploaded_date INTEGER DEFAULT (STRFTIME('%s', 'now')),"
"delete_date INTEGER)") "delete_date INTEGER)")
# init cron thread
t = CronThread()
t.start()
app.config["CRON_THREAD"] = t
def add_user(username, password, admin="FALSE"): def add_user(username, password, admin="FALSE"):
""" """
@ -429,7 +425,7 @@ def get_rand_chars(n):
init() init()
atexit.register(app.config["CRON_THREAD"].stop.set) atexit.register(scheduler.shutdown)
if __name__ == "__main__": if __name__ == "__main__":
import sys import sys
if len(sys.argv) > 1: if len(sys.argv) > 1: