Adding categories import

api/categories/controllers.py

@@ -32,4 +32,4 @@ def post(self):
     def get(self):
         categories = CategoriesModel.query.all()
         _categories = [category.to_dict() for category in categories]
-        return make_response(jsonify({'categories': _categories}), 200)
+        return make_response(jsonify({'categories': _categories}), 200)

app/__init__.py

@@ -8,6 +8,8 @@
 # Baseline
 from app.config import Config
 from app.database import db
+# CSV
+import csv
 
 def create_app():
     app = Flask(__name__)
@@ -54,6 +56,13 @@ def create_app():
         from api.categories.controllers import Categories
         from api.transaction.controllers import Transaction
 
+        #CLI
+        from app.cli import insert_categories
+        @app.cli.command('insert-categories')    
+        def insert_cat():
+            insert_categories()
+
+
         db.create_all()
 
     @login_manager.user_loader

app/cli.py

@@ -0,0 +1,53 @@
+# APP
+from app import db
+from app.config import Config
+# CSV
+import csv
+# MODELS
+from api.categories_type.models import CategoriesTypeModel as CategoriesType
+from api.categories_group.models import CategoriesGroupModel as CategoriesGroup
+from api.categories.models import CategoriesModel as Categories
+
+def insert_categories():
+    user_id = Config.DEFAULT_USER_ID
+    categories_type = []
+    categories_group = []
+    with open("data/categories_data.csv", "r") as f_in:
+        reader = csv.reader(f_in, quotechar="'")
+        next(reader)  # skip header
+        ## categories_type
+        for row in reader:
+            if row[2] not in categories_type:
+                categories_type.append(row[2])
+        ## categories_group
+            if row[1] not in categories_group:
+                categories_group.append(row[1])
+
+    for category_type in categories_type:
+        _categories_type = CategoriesType.query.filter_by(name=category_type, user_id=user_id).first()
+        if _categories_type is None:
+            cat = CategoriesType(name=category_type, user_id=user_id)
+            db.session.add(cat)
+            db.session.commit()
+    for category_group in categories_group:
+        _categories_group = CategoriesGroup.query.filter_by(name=category_group, user_id=user_id).first()
+        if _categories_group is None:
+            cat = CategoriesGroup(name=category_group, user_id=user_id)
+            db.session.add(cat)
+            db.session.commit()
+
+    with open("data/categories_data.csv", "r") as f_in:
+        reader = csv.reader(f_in, quotechar="'")
+        next(reader)  # skip header
+        ## categories_type
+        for row in reader:
+            category = row[0]
+            category_group = row[1]
+            category_type = row[2]
+            _category_type_id = CategoriesType.query.filter_by(name=category_type, user_id=user_id).first().id
+            _category_group_id = CategoriesGroup.query.filter_by(name=category_group, user_id=user_id).first().id
+            _categories = Categories.query.filter_by(name=category, categories_group_id=_category_group_id, categories_type_id=_category_type_id, user_id=user_id).first()
+            if _categories is None:
+                cat = Categories(name=category, categories_group_id=_category_group_id, categories_type_id=_category_type_id, user_id=user_id)
+                db.session.add(cat)
+                db.session.commit()

data/categories_data.csv

@@ -0,0 +1,131 @@
+categories,categories_group,categories_type
+Salary,Income,Income
+Bonus,Income,Income
+Dividend Income,Income,Income
+Interest Income,Income,Income
+Auto Insurance,Auto & Transportation,Expense
+Auto Payment,Auto & Transportation,Expense
+Gas & Fuel,Auto & Transportation,Expense
+Parking,Auto & Transportation,Expense
+Public Transportation,Auto & Transportation,Expense
+Service & Parts,Auto & Transportation,Expense
+Auto Registration,Auto & Transportation,Expense
+Quads,Auto & Transportation,Expense
+Tolls,Auto & Transportation,Expense
+RV Payment,Auto & Transportation,Expense
+ATV Insurance,Auto & Transportation,Expense
+Gardener,Bills & Utilities,Expense
+Landscape,Bills & Utilities,Expense
+Home Phone,Bills & Utilities,Expense
+Internet,Bills & Utilities,Expense
+Mobile Phone,Bills & Utilities,Expense
+Television,Bills & Utilities,Expense
+Utilities,Bills & Utilities,Expense
+Electric,Bills & Utilities,Expense
+Gas,Bills & Utilities,Expense
+Water and Trash,Bills & Utilities,Expense
+Pool,Bills & Utilities,Expense
+RV Insurance,Auto & Transportation,Expense
+Advertising,Business Services,Expense
+Legal,Business Services,Expense
+Office Supplies,Business Services,Expense
+Printing,Business Services,Expense
+Shipping,Business Services,Expense
+AWS Hosting,Business Services,Expense
+UTV Payment,Auto & Transport,Expense
+Books & Supplies,Education,Expense
+Student Loan,Education,Expense
+Tuition,Education,Expense
+Coaching,Education,Expense
+ATV Upgrads,Auto & Transportation,Expense
+Amusement,Entertainment,Expense
+Arts,Entertainment,Expense
+Movies & DVDs (AppleTV),Entertainment,Expense
+Music (Spotify),Entertainment,Expense
+Newspapers & Magazines,Entertainment,Expense
+Streaming (Netflix/Hulu/Twitch),Entertainment,Expense
+Subscriptions,Entertainment,Expense
+ATM Fee,Fees & Charges,Expense
+Credit Union Fee,Fees & Charges,Expense
+Finance Charge,Fees & Charges,Expense
+Late Fee,Fees & Charges,Expense
+Service Fee,Fees & Charges,Expense
+Trade Commissions,Fees & Charges,Expense
+Drinking Water,Food & Dining,Expense
+Financial Advisor,Financial,Expense
+Life Insurance,Financial,Expense
+Retirement,Financial,Expense
+Alcohol & Bars,Food & Dining,Expense
+Coffee Shops,Food & Dining,Expense
+Fast Food,Food & Dining,Expense
+Groceries,Food & Dining,Expense
+Restaurants,Food & Dining,Expense
+Monster,Food & Dining,Expense
+Charity,Gifts & Donations,Expense
+Gift,Gifts & Donations,Expense
+Dentist,Health & Fitness,Expense
+Doctor,Health & Fitness,Expense
+Eyecare,Health & Fitness,Expense
+Gym,Health & Fitness,Expense
+Health Insurance,Health & Fitness,Expense
+Pharmacy,Health & Fitness,Expense
+Sports,Health & Fitness,Expense
+Personal Care,Health & Fitness,Expense
+Furnishings,Home,Expense
+Home Improvement,Home,Expense
+Home Insurance,Home,Expense
+Home Services,Home,Expense
+Home Supplies,Home,Expense
+Lawn & Garden,Home,Expense
+Mortgage & Rent,Home,Expense
+Allowance,Kids,Expense
+Baby Supplies,Kids,Expense
+Babysitter & Daycare,Kids,Expense
+Child Support,Kids,Expense
+Kids Activities,Kids,Expense
+Toys,Kids,Expense
+Kids Clothing,Kids,Expense
+Hair,Personal Care,Expense
+Laundry,Personal Care,Expense
+Nail Salon,Personal Care,Expense
+Spa & Massage,Personal Care,Expense
+Tattoo,Personal Care,Expense
+Pet Food & Supplies,Pets,Expense
+Pet Grooming,Pets,Expense
+Veterinary,Pets,Expense
+Pet Insurance,Pets,Expense
+Adult Clothing,Personal Care,Expense
+Electronics & Software,Personal Care,Expense
+Hobbies,Personal Care,Expense
+Sporting Goods,Personal Care,Expense
+Federal Tax,Taxes,Expense
+Local Tax,Taxes,Expense
+Property Tax,Taxes,Expense
+Sales Tax,Taxes,Expense
+State Tax,Taxes,Expense
+Credit Card Payment,Transfer,Transfer
+Transfer for Cash Spending,Transfer,Transfer
+Transfer to Savings,Transfer,Transfer
+Transfer,Transfer,Transfer
+Air Travel,Travel,Expense
+Car Rental & Taxi,Travel,Expense
+Hotel,Travel,Expense
+Vacation,Travel,Expense
+Camping,Travel,Expense
+Passports,Travel,Expense
+Jocelyn DC Trip,Travel,Expense
+Cash & ATM,Uncategorized,Transfer
+Check,Uncategorized,Transfer
+Vehicle Sales,Auto & Transportation,Expense
+Drivers Licence,Auto & Transportation,Expense
+Donation,Gifts & Donations,Expense
+Tools,Home,Expense
+Home Escrow,Home,Expense
+RV Supplies,Auto & Transportation,Expense
+Vehicle Interest,Auto & Transportation,Expense
+Home Interest,Home,Expense
+Books,Education,Expense
+ADU,Home,Expense
+Photos,Home,Expense
+Education,Personal Care, Expense
+Assistant,Home,Expense