refactored app structure
This commit is contained in:
Bryan Bailey
147
web.py
147
web.py
@@ -1,141 +1,22 @@
|
||||
import os
|
||||
import threading
|
||||
from flask import Flask, render_template, url_for, request, flash, make_response, redirect, send_from_directory
|
||||
from flask_moment import Moment
|
||||
from flask_sqlalchemy import SQLAlchemy
|
||||
from flask_wtf import FlaskForm
|
||||
from wtforms import StringField, SubmitField, BooleanField
|
||||
from wtforms.validators import DataRequired
|
||||
|
||||
from flask_migrate import Migrate
|
||||
|
||||
from app import create_app, db
|
||||
from app.models import User, Role, ComicMeta
|
||||
from yoink.config import config
|
||||
from yoink.comic import Comic
|
||||
|
||||
|
||||
class DownloadForm(FlaskForm):
|
||||
url = StringField('Comic URL', validators=[DataRequired()])
|
||||
series = BooleanField('Series? ')
|
||||
download = SubmitField('Download')
|
||||
app = create_app(os.environ.get('FLASK_CONFIG') or 'default')
|
||||
migrate = Migrate(app, db)
|
||||
|
||||
app = Flask(__name__)
|
||||
moment = Moment(app)
|
||||
app.config['SECRET_KEY'] = 'snapekilleddumpledork'
|
||||
app.config['SQLALCHEMY_DATABASE_URI'] = f'sqlite:///{os.path.join(config.app_root, "data.sqlite")}'
|
||||
app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False
|
||||
db = SQLAlchemy(app)
|
||||
@app.shell_context_processor
|
||||
def make_shell_ctx(): return dict(db=db, User=User, Role=Role, ComicMeta=ComicMeta)
|
||||
|
||||
|
||||
|
||||
class Role(db.Model):
|
||||
__tablename__ = 'roles'
|
||||
id = db.Column(db.Integer, primary_key=True)
|
||||
name = db.Column(db.String(64), unique=True)
|
||||
users = db.relationship('User', backref='role', lazy='dynamic')
|
||||
|
||||
def __repr__(self):
|
||||
return f'<Role {self.name!r}>'
|
||||
|
||||
|
||||
class User(db.Model):
|
||||
__tablename__ = 'users'
|
||||
id = db.Column(db.Integer, primary_key=True)
|
||||
username = db.Column(db.String(64), unique=True, index=True)
|
||||
role_id = db.Column(db.Integer, db.ForeignKey('roles.id'))
|
||||
|
||||
def __repr__(self): return f'<User {self.username!r}>'
|
||||
|
||||
class ComicMeta(db.Model):
|
||||
__tablename__ = 'comicmeta'
|
||||
id = db.Column(db.Integer, primary_key=True)
|
||||
title = db.Column(db.String(256), unique=True, index=True)
|
||||
issue = db.Column(db.Integer, nullable=True, index=True)
|
||||
category = db.Column(db.String(128), index=True, nullable=True)
|
||||
previous_issue = db.Column(db.String(256), nullable=True)
|
||||
next_issue = db.Column(db.String(256), nullable=True)
|
||||
cover_path = db.Column(db.String(256))
|
||||
archive_path = db.Column(db.String(256))
|
||||
|
||||
|
||||
@app.errorhandler(404)
|
||||
def not_found(e):
|
||||
return render_template('404.html'), 404
|
||||
|
||||
@app.errorhandler(500)
|
||||
def server_error(e):
|
||||
return render_template('500.html'), 500
|
||||
|
||||
@app.route('/uploads/<path:filename>')
|
||||
def download_file(filename):
|
||||
return send_from_directory(os.path.join(config.library_path, 'comics'), filename, as_attachment=True)
|
||||
|
||||
|
||||
def get_cover_path(comic):
|
||||
return [image for image in os.listdir(os.path.join(config.library_path, 'comics', comic.title)) if image.endswith('000.jpg')][0]
|
||||
|
||||
|
||||
def get_archive_path(comic):
|
||||
return [image for image in os.listdir(os.path.join(config.library_path, 'comics', comic.title)) if image.endswith('.cbr')][0]
|
||||
|
||||
|
||||
def get_comic_library_meta():
|
||||
comic_meta = []
|
||||
|
||||
for comic in ComicMeta.query.all():
|
||||
comic_meta.append({
|
||||
'cover': comic.cover_path,
|
||||
'title': comic.title,
|
||||
'archive': comic.archive_path
|
||||
})
|
||||
|
||||
return comic_meta
|
||||
|
||||
|
||||
@app.route('/', methods=['post','get'])
|
||||
def index():
|
||||
url = None
|
||||
series = False
|
||||
form = DownloadForm()
|
||||
latest = get_comic_library_meta()
|
||||
|
||||
if form.validate_on_submit():
|
||||
url = form.url.data.strip()
|
||||
series = form.series.data
|
||||
|
||||
comic = Comic(url)
|
||||
comic_meta = ComicMeta.query.filter_by(title=comic.title).first()
|
||||
|
||||
comic.archiver.download()
|
||||
comic.archiver.generate_archive()
|
||||
|
||||
|
||||
if comic_meta is None:
|
||||
comic_meta = ComicMeta()
|
||||
comic_meta.title = comic.title
|
||||
comic_meta.category = comic.category
|
||||
comic_meta.issue = comic.issue_number
|
||||
comic_meta.next_issue = comic.next
|
||||
comic_meta.previous_issue = comic.prev
|
||||
comic_meta.cover_path = os.path.join(comic.title, get_cover_path(comic))
|
||||
comic_meta.archive_path = os.path.join(comic.title, get_archive_path(comic))
|
||||
|
||||
db.session.add(comic_meta)
|
||||
db.session.commit()
|
||||
else:
|
||||
flash(f'Comic {comic.title} exists')
|
||||
|
||||
latest = get_comic_library_meta()
|
||||
form.url.data = ''
|
||||
|
||||
return render_template('index.html', form=form, url=url, series=series, latest=latest), 200
|
||||
|
||||
|
||||
if form.series.data:
|
||||
print('Download the whole damn lot')
|
||||
|
||||
flash(f'{comic.title} downloaded to {os.path.join(config.library_path, "comics/" + comic.title)}')
|
||||
|
||||
latest = get_comic_library_meta()
|
||||
comic.archiver.cleanup_worktree()
|
||||
form.url.data = ''
|
||||
|
||||
return render_template('index.html', form=form, url=url, series=series, latest=latest), 200
|
||||
|
||||
@app.cli.command()
|
||||
def test():
|
||||
''' run unit tests '''
|
||||
import unittest
|
||||
tests = unittest.TestLoader().discover('tests')
|
||||
unittest.TextTestRunner(verbosity=2).run(tests)
|
||||
Reference in New Issue
Block a user