mirror of
https://github.com/Screenly/Anthias.git
synced 2026-05-06 14:25:33 -04:00
Improved importability of server module.
This commit is contained in:
Viktor Petersson
602
server.py
602
server.py
@@ -127,319 +127,319 @@ def initiate_db():
|
||||
c.execute("CREATE TABLE assets (asset_id TEXT, name TEXT, uri TEXT, md5 TEXT, start_date TIMESTAMP, end_date TIMESTAMP, duration TEXT, mimetype TEXT)")
|
||||
return "Initiated database."
|
||||
|
||||
|
||||
@route('/process_asset', method='POST')
|
||||
def process_asset():
|
||||
conn = sqlite3.connect(settings.database, detect_types=sqlite3.PARSE_DECLTYPES)
|
||||
c = conn.cursor()
|
||||
|
||||
if (request.POST.get('name', '').strip() and
|
||||
request.POST.get('uri', '').strip() and
|
||||
request.POST.get('mimetype', '').strip()
|
||||
):
|
||||
|
||||
name = request.POST.get('name', '').decode('UTF-8')
|
||||
uri = request.POST.get('uri', '').strip()
|
||||
mimetype = request.POST.get('mimetype', '').strip()
|
||||
|
||||
# Make sure it's a valid resource
|
||||
uri_check = urlparse(uri)
|
||||
if not (uri_check.scheme == "http" or uri_check.scheme == "https"):
|
||||
header = "Ops!"
|
||||
message = "URL must be HTTP or HTTPS."
|
||||
return template('message', header=header, message=message)
|
||||
|
||||
if "image" in mimetype:
|
||||
file = req_get(uri, allow_redirects=True)
|
||||
else:
|
||||
file = req_head(uri, allow_redirects=True)
|
||||
|
||||
# Only proceed if fetch was successful.
|
||||
if file.status_code == 200:
|
||||
asset_id = md5(name + uri).hexdigest()
|
||||
|
||||
strict_uri = file.url
|
||||
|
||||
if __name__ == "__main__":
|
||||
@route('/process_asset', method='POST')
|
||||
def process_asset():
|
||||
conn = sqlite3.connect(settings.database, detect_types=sqlite3.PARSE_DECLTYPES)
|
||||
c = conn.cursor()
|
||||
|
||||
if (request.POST.get('name', '').strip() and
|
||||
request.POST.get('uri', '').strip() and
|
||||
request.POST.get('mimetype', '').strip()
|
||||
):
|
||||
|
||||
name = request.POST.get('name', '').decode('UTF-8')
|
||||
uri = request.POST.get('uri', '').strip()
|
||||
mimetype = request.POST.get('mimetype', '').strip()
|
||||
|
||||
# Make sure it's a valid resource
|
||||
uri_check = urlparse(uri)
|
||||
if not (uri_check.scheme == "http" or uri_check.scheme == "https"):
|
||||
header = "Ops!"
|
||||
message = "URL must be HTTP or HTTPS."
|
||||
return template('message', header=header, message=message)
|
||||
|
||||
if "image" in mimetype:
|
||||
resolution = Image.open(StringIO(file.content)).size
|
||||
file = req_get(uri, allow_redirects=True)
|
||||
else:
|
||||
resolution = "N/A"
|
||||
|
||||
if "video" in mimetype:
|
||||
duration = "N/A"
|
||||
|
||||
start_date = ""
|
||||
end_date = ""
|
||||
duration = ""
|
||||
|
||||
c.execute("INSERT INTO assets (asset_id, name, uri, start_date, end_date, duration, mimetype) VALUES (?,?,?,?,?,?,?)", (asset_id, name, uri, start_date, end_date, duration, mimetype))
|
||||
conn.commit()
|
||||
|
||||
header = "Yay!"
|
||||
message = "Added asset (" + asset_id + ") to the database."
|
||||
return template('message', header=header, message=message)
|
||||
|
||||
file = req_head(uri, allow_redirects=True)
|
||||
|
||||
# Only proceed if fetch was successful.
|
||||
if file.status_code == 200:
|
||||
asset_id = md5(name + uri).hexdigest()
|
||||
|
||||
strict_uri = file.url
|
||||
|
||||
if "image" in mimetype:
|
||||
resolution = Image.open(StringIO(file.content)).size
|
||||
else:
|
||||
resolution = "N/A"
|
||||
|
||||
if "video" in mimetype:
|
||||
duration = "N/A"
|
||||
|
||||
start_date = ""
|
||||
end_date = ""
|
||||
duration = ""
|
||||
|
||||
c.execute("INSERT INTO assets (asset_id, name, uri, start_date, end_date, duration, mimetype) VALUES (?,?,?,?,?,?,?)", (asset_id, name, uri, start_date, end_date, duration, mimetype))
|
||||
conn.commit()
|
||||
|
||||
header = "Yay!"
|
||||
message = "Added asset (" + asset_id + ") to the database."
|
||||
return template('message', header=header, message=message)
|
||||
|
||||
else:
|
||||
header = "Ops!"
|
||||
message = "Unable to fetch file."
|
||||
return template('message', header=header, message=message)
|
||||
else:
|
||||
header = "Ops!"
|
||||
message = "Unable to fetch file."
|
||||
message = "Invalid input."
|
||||
return template('message', header=header, message=message)
|
||||
else:
|
||||
header = "Ops!"
|
||||
message = "Invalid input."
|
||||
return template('message', header=header, message=message)
|
||||
|
||||
|
||||
@route('/process_schedule', method='POST')
|
||||
def process_schedule():
|
||||
conn = sqlite3.connect(settings.database, detect_types=sqlite3.PARSE_DECLTYPES)
|
||||
c = conn.cursor()
|
||||
|
||||
if (request.POST.get('asset', '').strip() and
|
||||
request.POST.get('start', '').strip() and
|
||||
request.POST.get('end', '').strip()
|
||||
):
|
||||
|
||||
asset_id = request.POST.get('asset', '').strip()
|
||||
input_start = request.POST.get('start', '').strip()
|
||||
input_end = request.POST.get('end', '').strip()
|
||||
|
||||
start_date = datetime.strptime(input_start, '%Y-%m-%d @ %H:%M')
|
||||
end_date = datetime.strptime(input_end, '%Y-%m-%d @ %H:%M')
|
||||
|
||||
query = c.execute("SELECT mimetype FROM assets WHERE asset_id=?", (asset_id,))
|
||||
asset_mimetype = c.fetchone()
|
||||
|
||||
if "image" or "web" in asset_mimetype:
|
||||
|
||||
|
||||
@route('/process_schedule', method='POST')
|
||||
def process_schedule():
|
||||
conn = sqlite3.connect(settings.database, detect_types=sqlite3.PARSE_DECLTYPES)
|
||||
c = conn.cursor()
|
||||
|
||||
if (request.POST.get('asset', '').strip() and
|
||||
request.POST.get('start', '').strip() and
|
||||
request.POST.get('end', '').strip()
|
||||
):
|
||||
|
||||
asset_id = request.POST.get('asset', '').strip()
|
||||
input_start = request.POST.get('start', '').strip()
|
||||
input_end = request.POST.get('end', '').strip()
|
||||
|
||||
start_date = datetime.strptime(input_start, '%Y-%m-%d @ %H:%M')
|
||||
end_date = datetime.strptime(input_end, '%Y-%m-%d @ %H:%M')
|
||||
|
||||
query = c.execute("SELECT mimetype FROM assets WHERE asset_id=?", (asset_id,))
|
||||
asset_mimetype = c.fetchone()
|
||||
|
||||
if "image" or "web" in asset_mimetype:
|
||||
try:
|
||||
duration = request.POST.get('duration', '').strip()
|
||||
except:
|
||||
header = "Ops!"
|
||||
message = "Duration missing. This is required for images and web-pages."
|
||||
return template('message', header=header, message=message)
|
||||
else:
|
||||
duration = "N/A"
|
||||
|
||||
c.execute("UPDATE assets SET start_date=?, end_date=?, duration=? WHERE asset_id=?", (start_date, end_date, duration, asset_id))
|
||||
conn.commit()
|
||||
|
||||
header = "Yes!"
|
||||
message = "Successfully scheduled asset."
|
||||
return template('message', header=header, message=message)
|
||||
|
||||
else:
|
||||
header = "Ops!"
|
||||
message = "Failed to process schedule."
|
||||
return template('message', header=header, message=message)
|
||||
|
||||
|
||||
@route('/update_asset', method='POST')
|
||||
def update_asset():
|
||||
conn = sqlite3.connect(settings.database, detect_types=sqlite3.PARSE_DECLTYPES)
|
||||
c = conn.cursor()
|
||||
|
||||
if (request.POST.get('asset_id', '').strip() and
|
||||
request.POST.get('name', '').strip() and
|
||||
request.POST.get('uri', '').strip() and
|
||||
request.POST.get('mimetype', '').strip()
|
||||
):
|
||||
|
||||
asset_id = request.POST.get('asset_id', '').strip()
|
||||
name = request.POST.get('name', '').decode('UTF-8')
|
||||
uri = request.POST.get('uri', '').strip()
|
||||
mimetype = request.POST.get('mimetype', '').strip()
|
||||
|
||||
try:
|
||||
duration = request.POST.get('duration', '').strip()
|
||||
except:
|
||||
header = "Ops!"
|
||||
message = "Duration missing. This is required for images and web-pages."
|
||||
return template('message', header=header, message=message)
|
||||
duration = None
|
||||
|
||||
try:
|
||||
input_start = request.POST.get('start', '')
|
||||
start_date = datetime.strptime(input_start, '%Y-%m-%d @ %H:%M')
|
||||
except:
|
||||
start_date = None
|
||||
|
||||
try:
|
||||
input_end = request.POST.get('end', '').strip()
|
||||
end_date = datetime.strptime(input_end, '%Y-%m-%d @ %H:%M')
|
||||
except:
|
||||
end_date = None
|
||||
|
||||
c.execute("UPDATE assets SET start_date=?, end_date=?, duration=?, name=?, uri=?, duration=?, mimetype=? WHERE asset_id=?", (start_date, end_date, duration, name, uri, duration, mimetype, asset_id))
|
||||
conn.commit()
|
||||
|
||||
header = "Yes!"
|
||||
message = "Successfully updated asset."
|
||||
return template('message', header=header, message=message)
|
||||
|
||||
else:
|
||||
duration = "N/A"
|
||||
|
||||
c.execute("UPDATE assets SET start_date=?, end_date=?, duration=? WHERE asset_id=?", (start_date, end_date, duration, asset_id))
|
||||
conn.commit()
|
||||
|
||||
header = "Yes!"
|
||||
message = "Successfully scheduled asset."
|
||||
return template('message', header=header, message=message)
|
||||
|
||||
else:
|
||||
header = "Ops!"
|
||||
message = "Failed to process schedule."
|
||||
return template('message', header=header, message=message)
|
||||
|
||||
|
||||
@route('/update_asset', method='POST')
|
||||
def update_asset():
|
||||
conn = sqlite3.connect(settings.database, detect_types=sqlite3.PARSE_DECLTYPES)
|
||||
c = conn.cursor()
|
||||
|
||||
if (request.POST.get('asset_id', '').strip() and
|
||||
request.POST.get('name', '').strip() and
|
||||
request.POST.get('uri', '').strip() and
|
||||
request.POST.get('mimetype', '').strip()
|
||||
):
|
||||
|
||||
asset_id = request.POST.get('asset_id', '').strip()
|
||||
name = request.POST.get('name', '').decode('UTF-8')
|
||||
uri = request.POST.get('uri', '').strip()
|
||||
mimetype = request.POST.get('mimetype', '').strip()
|
||||
|
||||
header = "Ops!"
|
||||
message = "Failed to update asset."
|
||||
return template('message', header=header, message=message)
|
||||
|
||||
|
||||
@route('/delete_asset/:asset_id')
|
||||
def delete_asset(asset_id):
|
||||
conn = sqlite3.connect(settings.database, detect_types=sqlite3.PARSE_DECLTYPES)
|
||||
c = conn.cursor()
|
||||
|
||||
c.execute("DELETE FROM assets WHERE asset_id=?", (asset_id,))
|
||||
try:
|
||||
duration = request.POST.get('duration', '').strip()
|
||||
conn.commit()
|
||||
|
||||
header = "Success!"
|
||||
message = "Deleted asset."
|
||||
return template('message', header=header, message=message)
|
||||
except:
|
||||
duration = None
|
||||
|
||||
header = "Ops!"
|
||||
message = "Failed to delete asset."
|
||||
return template('message', header=header, message=message)
|
||||
|
||||
|
||||
@route('/')
|
||||
def viewIndex():
|
||||
initiate_db()
|
||||
return template('index')
|
||||
|
||||
|
||||
@route('/system_info')
|
||||
def system_info():
|
||||
viewer_log_file = '/tmp/screenly_viewer.log'
|
||||
if path.exists(viewer_log_file):
|
||||
viewlog = check_output(['tail', '-n', '20', viewer_log_file]).split('\n')
|
||||
else:
|
||||
viewlog = ["(no viewer log present -- is only the screenly server running?)\n"]
|
||||
|
||||
loadavg = getloadavg()[2]
|
||||
|
||||
resolution = check_output(['tvservice', '-s']).strip()
|
||||
|
||||
# Calculate disk space
|
||||
slash = statvfs("/")
|
||||
free_space = size(slash.f_bsize * slash.f_bavail)
|
||||
|
||||
# Get uptime
|
||||
with open('/proc/uptime', 'r') as f:
|
||||
uptime_seconds = float(f.readline().split()[0])
|
||||
uptime = str(timedelta(seconds=uptime_seconds))
|
||||
|
||||
return template('system_info', viewlog=viewlog, loadavg=loadavg, free_space=free_space, uptime=uptime, resolution=resolution)
|
||||
|
||||
|
||||
@route('/splash_page')
|
||||
def splash_page():
|
||||
# Make sure the database exist and that it is initiated.
|
||||
initiate_db()
|
||||
|
||||
try:
|
||||
input_start = request.POST.get('start', '')
|
||||
start_date = datetime.strptime(input_start, '%Y-%m-%d @ %H:%M')
|
||||
my_ip = ifaddresses('eth0')[2][0]['addr']
|
||||
ip_lookup = True
|
||||
url = 'http://' + my_ip + ':8080'
|
||||
except:
|
||||
start_date = None
|
||||
|
||||
try:
|
||||
input_end = request.POST.get('end', '').strip()
|
||||
end_date = datetime.strptime(input_end, '%Y-%m-%d @ %H:%M')
|
||||
except:
|
||||
end_date = None
|
||||
|
||||
c.execute("UPDATE assets SET start_date=?, end_date=?, duration=?, name=?, uri=?, duration=?, mimetype=? WHERE asset_id=?", (start_date, end_date, duration, name, uri, duration, mimetype, asset_id))
|
||||
conn.commit()
|
||||
|
||||
header = "Yes!"
|
||||
message = "Successfully updated asset."
|
||||
return template('message', header=header, message=message)
|
||||
|
||||
else:
|
||||
header = "Ops!"
|
||||
message = "Failed to update asset."
|
||||
return template('message', header=header, message=message)
|
||||
|
||||
|
||||
@route('/delete_asset/:asset_id')
|
||||
def delete_asset(asset_id):
|
||||
conn = sqlite3.connect(settings.database, detect_types=sqlite3.PARSE_DECLTYPES)
|
||||
c = conn.cursor()
|
||||
|
||||
c.execute("DELETE FROM assets WHERE asset_id=?", (asset_id,))
|
||||
try:
|
||||
conn.commit()
|
||||
|
||||
header = "Success!"
|
||||
message = "Deleted asset."
|
||||
return template('message', header=header, message=message)
|
||||
except:
|
||||
header = "Ops!"
|
||||
message = "Failed to delete asset."
|
||||
return template('message', header=header, message=message)
|
||||
|
||||
|
||||
@route('/')
|
||||
def viewIndex():
|
||||
initiate_db()
|
||||
return template('index')
|
||||
|
||||
|
||||
@route('/system_info')
|
||||
def system_info():
|
||||
viewer_log_file = '/tmp/screenly_viewer.log'
|
||||
if path.exists(viewer_log_file):
|
||||
viewlog = check_output(['tail', '-n', '20', viewer_log_file]).split('\n')
|
||||
else:
|
||||
viewlog = ["(no viewer log present -- is only the screenly server running?)\n"]
|
||||
|
||||
loadavg = getloadavg()[2]
|
||||
|
||||
resolution = check_output(['tvservice', '-s']).strip()
|
||||
|
||||
# Calculate disk space
|
||||
slash = statvfs("/")
|
||||
free_space = size(slash.f_bsize * slash.f_bavail)
|
||||
|
||||
# Get uptime
|
||||
with open('/proc/uptime', 'r') as f:
|
||||
uptime_seconds = float(f.readline().split()[0])
|
||||
uptime = str(timedelta(seconds=uptime_seconds))
|
||||
|
||||
return template('system_info', viewlog=viewlog, loadavg=loadavg, free_space=free_space, uptime=uptime, resolution=resolution)
|
||||
|
||||
|
||||
@route('/splash_page')
|
||||
def splash_page():
|
||||
# Make sure the database exist and that it is initiated.
|
||||
initiate_db()
|
||||
|
||||
try:
|
||||
my_ip = ifaddresses('eth0')[2][0]['addr']
|
||||
ip_lookup = True
|
||||
url = 'http://' + my_ip + ':8080'
|
||||
except:
|
||||
ip_lookup = False
|
||||
url = "Unable to lookup IP from eth0."
|
||||
|
||||
return template('splash_page', ip_lookup=ip_lookup, url=url)
|
||||
|
||||
|
||||
@route('/view_playlist')
|
||||
def view_node_playlist():
|
||||
|
||||
nodeplaylist = loads(get_playlist())
|
||||
|
||||
return template('view_playlist', nodeplaylist=nodeplaylist)
|
||||
|
||||
|
||||
@route('/view_assets')
|
||||
def view_assets():
|
||||
|
||||
nodeplaylist = loads(get_assets())
|
||||
|
||||
return template('view_assets', nodeplaylist=nodeplaylist)
|
||||
|
||||
|
||||
@route('/add_asset')
|
||||
def add_asset():
|
||||
return template('add_asset')
|
||||
|
||||
|
||||
@route('/schedule_asset')
|
||||
def schedule_asset():
|
||||
conn = sqlite3.connect(settings.database, detect_types=sqlite3.PARSE_DECLTYPES)
|
||||
c = conn.cursor()
|
||||
|
||||
assets = []
|
||||
c.execute("SELECT name, asset_id FROM assets ORDER BY name")
|
||||
query = c.fetchall()
|
||||
for asset in query:
|
||||
ip_lookup = False
|
||||
url = "Unable to lookup IP from eth0."
|
||||
|
||||
return template('splash_page', ip_lookup=ip_lookup, url=url)
|
||||
|
||||
|
||||
@route('/view_playlist')
|
||||
def view_node_playlist():
|
||||
|
||||
nodeplaylist = loads(get_playlist())
|
||||
|
||||
return template('view_playlist', nodeplaylist=nodeplaylist)
|
||||
|
||||
|
||||
@route('/view_assets')
|
||||
def view_assets():
|
||||
|
||||
nodeplaylist = loads(get_assets())
|
||||
|
||||
return template('view_assets', nodeplaylist=nodeplaylist)
|
||||
|
||||
|
||||
@route('/add_asset')
|
||||
def add_asset():
|
||||
return template('add_asset')
|
||||
|
||||
|
||||
@route('/schedule_asset')
|
||||
def schedule_asset():
|
||||
conn = sqlite3.connect(settings.database, detect_types=sqlite3.PARSE_DECLTYPES)
|
||||
c = conn.cursor()
|
||||
|
||||
assets = []
|
||||
c.execute("SELECT name, asset_id FROM assets ORDER BY name")
|
||||
query = c.fetchall()
|
||||
for asset in query:
|
||||
name = asset[0]
|
||||
asset_id = asset[1]
|
||||
|
||||
assets.append({
|
||||
'name': name,
|
||||
'asset_id': asset_id,
|
||||
})
|
||||
|
||||
return template('schedule_asset', assets=assets)
|
||||
|
||||
|
||||
@route('/edit_asset/:asset_id')
|
||||
def edit_asset(asset_id):
|
||||
conn = sqlite3.connect(settings.database, detect_types=sqlite3.PARSE_DECLTYPES)
|
||||
c = conn.cursor()
|
||||
|
||||
c.execute("SELECT name, uri, md5, start_date, end_date, duration, mimetype FROM assets WHERE asset_id=?", (asset_id,))
|
||||
asset = c.fetchone()
|
||||
|
||||
name = asset[0]
|
||||
asset_id = asset[1]
|
||||
|
||||
assets.append({
|
||||
'name': name,
|
||||
'asset_id': asset_id,
|
||||
})
|
||||
|
||||
return template('schedule_asset', assets=assets)
|
||||
|
||||
|
||||
@route('/edit_asset/:asset_id')
|
||||
def edit_asset(asset_id):
|
||||
conn = sqlite3.connect(settings.database, detect_types=sqlite3.PARSE_DECLTYPES)
|
||||
c = conn.cursor()
|
||||
|
||||
c.execute("SELECT name, uri, md5, start_date, end_date, duration, mimetype FROM assets WHERE asset_id=?", (asset_id,))
|
||||
asset = c.fetchone()
|
||||
|
||||
name = asset[0]
|
||||
uri = asset[1]
|
||||
md5 = asset[2]
|
||||
|
||||
if asset[3]:
|
||||
start_date = datestring.date_to_string(asset[3])
|
||||
uri = asset[1]
|
||||
md5 = asset[2]
|
||||
|
||||
if asset[3]:
|
||||
start_date = datestring.date_to_string(asset[3])
|
||||
else:
|
||||
start_date = None
|
||||
|
||||
if asset[4]:
|
||||
end_date = datestring.date_to_string(asset[4])
|
||||
else:
|
||||
end_date = None
|
||||
|
||||
duration = asset[5]
|
||||
mimetype = asset[6]
|
||||
|
||||
asset_info = {
|
||||
"name": name,
|
||||
"uri": uri,
|
||||
"duration": duration,
|
||||
"mimetype": mimetype,
|
||||
"asset_id": asset_id,
|
||||
"start_date": start_date,
|
||||
"end_date": end_date
|
||||
}
|
||||
#return str(asset_info)
|
||||
return template('edit_asset', asset_info=asset_info)
|
||||
|
||||
# Static
|
||||
|
||||
|
||||
@route('/static/:path#.+#', name='static')
|
||||
def static(path):
|
||||
return static_file(path, root='static')
|
||||
|
||||
|
||||
@error(403)
|
||||
def mistake403(code):
|
||||
return 'The parameter you passed has the wrong format!'
|
||||
|
||||
|
||||
@error(404)
|
||||
def mistake404(code):
|
||||
return 'Sorry, this page does not exist!'
|
||||
|
||||
# Ugly local dev fix.
|
||||
if platform == "darwin":
|
||||
port = '8080'
|
||||
run(host='127.0.0.1', port=port, reloader=True)
|
||||
else:
|
||||
start_date = None
|
||||
|
||||
if asset[4]:
|
||||
end_date = datestring.date_to_string(asset[4])
|
||||
else:
|
||||
end_date = None
|
||||
|
||||
duration = asset[5]
|
||||
mimetype = asset[6]
|
||||
|
||||
asset_info = {
|
||||
"name": name,
|
||||
"uri": uri,
|
||||
"duration": duration,
|
||||
"mimetype": mimetype,
|
||||
"asset_id": asset_id,
|
||||
"start_date": start_date,
|
||||
"end_date": end_date
|
||||
}
|
||||
#return str(asset_info)
|
||||
return template('edit_asset', asset_info=asset_info)
|
||||
|
||||
# Static
|
||||
|
||||
|
||||
@route('/static/:path#.+#', name='static')
|
||||
def static(path):
|
||||
return static_file(path, root='static')
|
||||
|
||||
|
||||
@error(403)
|
||||
def mistake403(code):
|
||||
return 'The parameter you passed has the wrong format!'
|
||||
|
||||
|
||||
@error(404)
|
||||
def mistake404(code):
|
||||
return 'Sorry, this page does not exist!'
|
||||
|
||||
# Ugly local dev fix.
|
||||
if platform == "darwin":
|
||||
port = '8080'
|
||||
run(host='127.0.0.1', port=port, reloader=True)
|
||||
else:
|
||||
run(host='0.0.0.0', port=8080, reloader=True)
|
||||
run(host='0.0.0.0', port=8080, reloader=True)
|
||||
|
||||
Reference in New Issue
Block a user