Los archivos se guardan en la base de datos y se agregaron categorias
Si, creo que eso nada mas
This commit is contained in:
Daniel Cortes
195
files/files.py
195
files/files.py
@@ -1,12 +1,13 @@
|
||||
import os
|
||||
import random
|
||||
|
||||
from flask import Flask, Blueprint, flash, request, redirect, url_for, current_app, render_template, send_from_directory
|
||||
from flask import Flask, Blueprint, flash, request, redirect, url_for, current_app, render_template, send_from_directory, g
|
||||
|
||||
from werkzeug.utils import secure_filename
|
||||
from werkzeug.exceptions import abort
|
||||
|
||||
from files.auth import admin_required
|
||||
from files.db import get_db
|
||||
|
||||
|
||||
bp = Blueprint('files', __name__)
|
||||
@@ -18,62 +19,194 @@ def get_extension(filename):
|
||||
def get_path_in_upload(filename):
|
||||
return os.path.join(current_app.config['UPLOAD_FOLDER'], filename)
|
||||
|
||||
def save_file(file, private, category):
|
||||
if private is None:
|
||||
is_private = 0
|
||||
else:
|
||||
is_private = 1
|
||||
|
||||
db = get_db()
|
||||
|
||||
filename = secure_filename(file.filename)
|
||||
db.execute(
|
||||
'INSERT INTO files (filename, private, category)'
|
||||
' VALUES (?,?,?)',
|
||||
(filename, is_private, category['id'],)
|
||||
)
|
||||
|
||||
file.save(get_path_in_upload(filename))
|
||||
|
||||
db.commit()
|
||||
|
||||
return filename
|
||||
|
||||
def _rename_file(file, new_name):
|
||||
db = get_db()
|
||||
|
||||
new_name = secure_filename(new_name)
|
||||
db.execute(
|
||||
'UPDATE files'
|
||||
' SET filename = ?'
|
||||
' WHERE id = ?',
|
||||
(new_name, file['id'],)
|
||||
)
|
||||
|
||||
os.rename(get_path_in_upload(file['filename']), get_path_in_upload(new_name))
|
||||
|
||||
db.commit()
|
||||
|
||||
return new_name
|
||||
|
||||
|
||||
def _delete_file(file):
|
||||
db = get_db()
|
||||
db.execute(
|
||||
'DELETE from files'
|
||||
' WHERE id = ?',
|
||||
(file['id'],)
|
||||
)
|
||||
|
||||
os.remove(get_path_in_upload(file['filename']))
|
||||
db.commit()
|
||||
|
||||
|
||||
def get_files():
|
||||
db = get_db()
|
||||
files = db.execute(
|
||||
'SELECT id, filename, private, category'
|
||||
' FROM files'
|
||||
' ORDER BY filename'
|
||||
).fetchall()
|
||||
return files
|
||||
|
||||
def get_public_files():
|
||||
db = get_db()
|
||||
files = db.execute(
|
||||
'SELECT id, filename, private, category'
|
||||
' FROM files'
|
||||
' WHERE private = false'
|
||||
' ORDER BY filename'
|
||||
).fetchall()
|
||||
|
||||
return files
|
||||
|
||||
def get_files_by_category(id):
|
||||
db = get_db()
|
||||
files = db.execute(
|
||||
'SELECT id, filename, private'
|
||||
' FROM files'
|
||||
' WHERE category = ?'
|
||||
' ORDER BY filename',
|
||||
(id,)
|
||||
).fetchall()
|
||||
|
||||
def get_file(id):
|
||||
db = get_db()
|
||||
file = db.execute(
|
||||
'SELECT id, filename, private, category'
|
||||
' FROM files'
|
||||
' WHERE id = ?'
|
||||
' LIMIT 1',
|
||||
(id,)
|
||||
).fetchone()
|
||||
return file
|
||||
|
||||
def get_categories():
|
||||
db = get_db()
|
||||
categories = db.execute(
|
||||
'SELECT id, name'
|
||||
' FROM categories'
|
||||
' ORDER BY name'
|
||||
).fetchall()
|
||||
return categories
|
||||
|
||||
def get_category(id):
|
||||
db = get_db()
|
||||
category = db.execute(
|
||||
'SELECT id, name'
|
||||
' FROM categories'
|
||||
' WHERE id = ?'
|
||||
' LIMIT 1',
|
||||
(id,)
|
||||
).fetchone()
|
||||
return category
|
||||
|
||||
@bp.route('/', methods=['GET', 'POST'])
|
||||
def index():
|
||||
filenames = os.listdir(current_app.config['UPLOAD_FOLDER'])
|
||||
return render_template('files/index.html', filenames=filenames)
|
||||
if g.user is None:
|
||||
files = get_public_files()
|
||||
else:
|
||||
files = get_files()
|
||||
|
||||
return render_template('files/index.html', files=files)
|
||||
|
||||
@bp.route('/upload', methods=['GET', 'POST'])
|
||||
@admin_required
|
||||
def upload_file():
|
||||
db = get_db()
|
||||
|
||||
if request.method == 'POST':
|
||||
if 'file' not in request.files:
|
||||
flash('No file part')
|
||||
return redirect(request.url)
|
||||
file = request.files['file']
|
||||
else:
|
||||
file = request.files['file']
|
||||
|
||||
if 'private' not in request.form:
|
||||
private = None
|
||||
else:
|
||||
private = request.form['private']
|
||||
|
||||
if 'category' not in request.form:
|
||||
flash('No category selected')
|
||||
return redirect(request.url)
|
||||
else:
|
||||
category = get_category(request.form['category'])
|
||||
|
||||
if file.filename == '':
|
||||
flash('No seleted file')
|
||||
return rediret(request.url)
|
||||
return redirect(request.url)
|
||||
|
||||
if file:
|
||||
filename = secure_filename(file.filename)
|
||||
file.save(get_path_in_upload(filename))
|
||||
filename = save_file(file, private, category)
|
||||
return redirect(url_for('files.index'))
|
||||
|
||||
return render_template('files/upload.html', categories=get_categories())
|
||||
|
||||
return redirect(url_for('files.preview_file', filename=filename))
|
||||
return render_template('files/upload.html')
|
||||
|
||||
@bp.route('/preview/<path:filename>')
|
||||
def preview_file(filename):
|
||||
file = os.path.isfile(get_path_in_upload(filename))
|
||||
if os.path.isfile(get_path_in_upload(filename)):
|
||||
return render_template('files/preview.html', filename=filename)
|
||||
@bp.route('/preview/<int:id>')
|
||||
def preview_file(id):
|
||||
file = get_file(id)
|
||||
if (file['private'] == 1 and g.user is not None) or (file['private'] == 0):
|
||||
return render_template('files/preview.html', file=file)
|
||||
else:
|
||||
abort(404)
|
||||
return abort(404)
|
||||
|
||||
|
||||
@bp.route('/rename/<path:filename>', methods=['POST'])
|
||||
@bp.route('/rename/<int:id>', methods=['POST'])
|
||||
@admin_required
|
||||
def rename_file(filename):
|
||||
if request.method == 'POST' and os.path.isfile(get_path_in_upload(filename)):
|
||||
def rename_file(id):
|
||||
if request.method == 'POST':
|
||||
file = get_file(id)
|
||||
|
||||
new_name = request.form['new_name'].lower()
|
||||
extension = filename.rsplit('.', 1)[1].lower()
|
||||
extension = file['filename'].rsplit('.', 1)[1].lower()
|
||||
|
||||
if "." in new_name and get_extension(new_name):
|
||||
new_name = new_name.rsplit('.',1)[0] + '.' + extension
|
||||
else:
|
||||
new_name = new_name + '.' + extension
|
||||
|
||||
_rename_file(file, new_name)
|
||||
|
||||
return redirect(url_for('files.preview_file', id=file['id']))
|
||||
|
||||
new_name = secure_filename(new_name)
|
||||
os.rename(get_path_in_upload(filename), get_path_in_upload(new_name))
|
||||
|
||||
return redirect(url_for('files.preview_file', filename=new_name))
|
||||
|
||||
@bp.route('/delete/<path:filename>', methods=['POST'])
|
||||
@bp.route('/delete/<int:id>', methods=['POST'])
|
||||
@admin_required
|
||||
def delete_file(filename):
|
||||
full_path = os.path.join(current_app.config['UPLOAD_FOLDER'], filename)
|
||||
|
||||
if request.method == 'POST' and os.path.isfile(full_path):
|
||||
os.remove(full_path)
|
||||
def delete_file(id):
|
||||
if request.method == 'POST':
|
||||
file = get_file(id)
|
||||
print(file)
|
||||
_delete_file(file)
|
||||
return redirect(url_for('index'))
|
||||
else:
|
||||
abort(404)
|
||||
|
||||
Reference in New Issue
Block a user