def _get_ufile(file_indicator = None):
ufile = UFile.query.filter(UFile.url == file_indicator).first()
if not ufile:
ufile = None
if ufile is not None:
# test whether file exists
if not funcs.f_exists(ufile.filename):
db_session.delete(ufile)
db_session.commit()
abort(404)
else:
if not funcs.f_exists(file_indicator):
abort(404)
ufile = UFile(
name = funcs.get_name_from_filename(file_indicator),
url = file_indicator,
filename = file_indicator,
filesize = funcs.get_file_size(file_indicator),
mimetype = funcs.get_file_mimetype(file_indicator),
created = datetime.utcnow()
)
db_session.add(ufile)
db_session.commit()
#if not r:
# logging.error('Failed to commit new record: %r' % ufile)
# abort(404)
return ufile
def gen_file_item(realfile, o_filename):
pos = o_filename.rfind(".")
if pos != -1:
ufile_name = o_filename[:pos]
else:
ufile_name = o_filename
d_fn = os.path.basename(realfile)
ufile = UFile(
name=ufile_name,
url=d_fn,
filename=d_fn,
filesize=funcs.get_file_size(d_fn),
mimetype=funcs.get_file_mimetype(d_fn),
created=datetime.utcnow(),
)
db_session.add(ufile)
db_session.commit()
return d_fn
def edit_file(file_indicator = None):
if request.method == 'GET' and file_indicator is None: return homepage()
if not funcs.is_admin_login():
abort(403)
ufile = None
if request.method == 'POST':
try:
ufile = UFile.query.filter(UFile.id == int(request.form.get('file_id'))).first()
except:
ufile = None
if not ufile:
abort(403)
err = False
name = Markup(request.form.get('name')).striptags().strip()
if not name:
err = True
flash('Title is empty or contains illegal characters.')
else:
ufile.name = name
url = re.sub('[^%a-zA-Z0-9_\-\.]', '', request.form.get('url')).strip('-')
if not url:
err = True
flash('URL is empty or contains illegal characters.')
else:
ufile.url = url
if UFile.query.filter(UFile.url == 'url').count() > 0:
err = True
flash('URL has already existed.')
ufile.password = request.form.get('password').strip()
if not ufile.password: ufile.password = None
ufile.description = request.form.get('description').strip()
if not ufile.description: ufile.description = None
try:
expire_delta = int(request.form.get('expire_delta'))
except:
expire_delta = -1
if expire_delta == -1:
pass
else:
if expire_delta == 0:
ufile.expire_at = None
else:
ufile.expire_at = datetime.utcnow()+timedelta(hours=expire_delta)
for item in ['linkable', 'download', 'homeshow']:
if request.form.get(item) == 'yes':
setattr(ufile, item, True)
else:
setattr(ufile, item, False)
if not err:
db_session.add(ufile)
try:
db_session.commit()
return redirect(url_for('file_serve',file_indicator=url))
except:
flash('Failed to update database.')
if ufile is None: ufile = _get_ufile(file_indicator)
preview = _get_preview(ufile)
fileext = _get_fileext(ufile)
visitkey = gen_passwd()
page_title = "Edit \"%s\"" % ufile.name
if ufile.description is None: ufile.description = ''
if ufile.password is None: ufile.password = ''
response = make_response(render_template('edit.html', ufile=ufile, preview = preview, fileext = fileext, file_indicator = file_indicator, visitkey=visitkey, edit_page = False, title=page_title))
response.set_cookie(sf_cookie_name(ufile.filename), sf_cookie_val(ufile.filename, visitkey))
return response
