From b147cfeb81baa5e06d3995f9611bcacec9aafeb0 Mon Sep 17 00:00:00 2001
From: Yara Naika <a@b.c>
Date: Sat, 13 May 2017 23:51:29 -0700
Subject: [PATCH] Ghetto migrate solution, in case you re-run db_create, it
 won't add categories again

---
 db_create.py | 51 ++++++++++++++++++++++++++-------------------------
 1 file changed, 26 insertions(+), 25 deletions(-)

diff --git a/db_create.py b/db_create.py
index e3a668c..99c4a85 100644
--- a/db_create.py
+++ b/db_create.py
@@ -5,33 +5,34 @@ from nyaa import app, db, models
 
 db.create_all()
 
-# Insert categories
+# 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 = []
 
-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)
 
-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()
+    db.session.commit()
 
 # Create fulltext index