63 lines
1.9 KiB
Python
63 lines
1.9 KiB
Python
import logging
|
|
from os import environ, urandom
|
|
|
|
from flask import Flask
|
|
from flask.cli import load_dotenv
|
|
from flask_httpauth import HTTPBasicAuth
|
|
from flask_sqlalchemy import SQLAlchemy
|
|
from redis import Redis
|
|
from transmission_rpc.client import Client
|
|
|
|
load_dotenv()
|
|
|
|
IS_DEBUG = environ.get('FLASK_ENV', 'production') == 'development'
|
|
ADMIN_USERNAME = environ.get('ADMIN_USERNAME', 'admin')
|
|
ADMIN_PASSWORD = environ.get('ADMIN_PASSWORD', 'secret')
|
|
APP_PORT = int(environ.get('FLASK_PORT', 5000))
|
|
CACHE_TIMEOUT = int(environ.get('CACHE_TIMEOUT', 60 * 60))
|
|
REQUESTS_TIMEOUT = int(environ.get('REQUESTS_TIMEOUT', 5))
|
|
BLACKLIST_WORDS = environ.get('BLACKLIST_WORDS', '').split(',') if environ.get('BLACKLIST_WORDS', '') else []
|
|
DB_ENABLED = False
|
|
REDIS_ENABLED = False
|
|
TRANSMISSION_ENABLED = False
|
|
|
|
app = Flask(__name__)
|
|
app.name = 'PyNyaaTa'
|
|
app.debug = IS_DEBUG
|
|
app.secret_key = urandom(24).hex()
|
|
app.url_map.strict_slashes = False
|
|
auth = HTTPBasicAuth()
|
|
logging.basicConfig(level=(logging.DEBUG if IS_DEBUG else logging.INFO))
|
|
logger = logging.getLogger(app.name)
|
|
|
|
db_uri = environ.get('DATABASE_URI')
|
|
if db_uri:
|
|
DB_ENABLED = True
|
|
app.config['SQLALCHEMY_DATABASE_URI'] = db_uri
|
|
app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = True
|
|
app.config['SQLALCHEMY_ECHO'] = IS_DEBUG
|
|
app.config['SQLALCHEMY_ENGINE_OPTIONS'] = {
|
|
'pool_recycle': 200
|
|
}
|
|
db = SQLAlchemy(app)
|
|
from .models import create_all
|
|
|
|
create_all()
|
|
|
|
cache_host = environ.get('REDIS_SERVER')
|
|
if cache_host:
|
|
REDIS_ENABLED = True
|
|
cache = Redis(cache_host)
|
|
|
|
transmission_host = environ.get('TRANSMISSION_SERVER')
|
|
if transmission_host:
|
|
TRANSMISSION_ENABLED = True
|
|
transmission_username = environ.get('TRANSMISSION_RPC_USERNAME')
|
|
transmission_password = environ.get('TRANSMISSION_RPC_PASSWORD')
|
|
transmission = Client(
|
|
username=transmission_username,
|
|
password=transmission_password,
|
|
host=transmission_host,
|
|
logger=logger
|
|
)
|