mirror of
https://gitlab.com/SIGBUS/nyaa.git
synced 2024-10-31 23:55:54 +00:00
152e547ac5
Update some dependencies to their latest version. Make executable scripts executable (chmod +x).
36 lines
1.3 KiB
Python
Executable file
36 lines
1.3 KiB
Python
Executable file
import sys
|
|
from nyaa import app, db, models
|
|
|
|
# Create tables
|
|
|
|
db.create_all()
|
|
|
|
# Insert categories and insert if it doesn't eixst
|
|
existing_cats = models.MainCategory.query.all()
|
|
if not existing_cats:
|
|
if app.config['SITE_FLAVOR'] == 'nyaa':
|
|
CATEGORIES = [
|
|
('Anime', ['Anime Music Video', 'English-translated', 'Non-English-translated', 'Raw']),
|
|
('Audio', ['Lossless', 'Lossy']),
|
|
('Literature', ['English-translated', 'Non-English-translated', 'Raw']),
|
|
('Live Action', ['English-translated', 'Idol/Promotional Video', 'Non-English-translated', 'Raw']),
|
|
('Pictures', ['Graphics', 'Photos']),
|
|
('Software', ['Applications', 'Games']),
|
|
]
|
|
elif app.config['SITE_FLAVOR'] == 'sukebei':
|
|
CATEGORIES = [
|
|
('Art', ['Anime', 'Doujinshi', 'Games', 'Manga', 'Pictures']),
|
|
('Real Life', ['Photobooks / Pictures', 'Videos']),
|
|
]
|
|
else:
|
|
CATEGORIES = []
|
|
|
|
for main_cat_name, sub_cat_names in CATEGORIES:
|
|
main_cat = models.MainCategory(name=main_cat_name)
|
|
for i, sub_cat_name in enumerate(sub_cat_names):
|
|
# Composite keys can't autoincrement, set sub_cat id manually (1-index)
|
|
sub_cat = models.SubCategory(id=i+1, name=sub_cat_name, main_category=main_cat)
|
|
db.session.add(main_cat)
|
|
|
|
db.session.commit()
|