skrd/flask-files
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Added filetype support

Browse Source
This commit is contained in:
Daniel Cortes
2019-03-12 00:51:29 -03:00
parent f12fe5b46d
commit 77c901f4f0
5 changed files with 97 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
files/commands.py
View File

@@ -6,7 +6,7 @@ from flask.cli import with_appcontext
from werkzeug.security import generate_password_hash
from files.models import db, User, Category, File
from files.models import db, User, Category, File, FileType
def init_db():
@@ -25,19 +25,26 @@ def generate_admin():
def add_defaults():
default_category = Category('Default')
default_file_type = FileType('Default')
code_file_type = FileType('Code')
image_file_type = FileType('Image')
db.session.add(default_category)
db.session.add(default_file_type)
db.session.add(code_file_type)
db.session.add(image_file_type)
db.session.commit()
def add_files():
existing_files = os.listdir(current_app.config['UPLOAD_FOLDER'])
default_category = Category.query.filter_by(name='Default').first()
default_file_type = FileType.query.filter_by(name='Default').first()
added = 0
for f in existing_files:
search = File.query.filter_by(filename=f).first()
if search is None:
file = File(f, 0, default_category.id)
file = File(f, 0, default_category.id, default_file_type.id)
db.session.add(file)
added += 1
db.session.commit()