141 lines
4.4 KiB
Python
141 lines
4.4 KiB
Python
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 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 = 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)
|
|
|
|
|
|
|
|
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
|
|
|