def category_create(user):
category_fields = category_schema.load(request.json)
new_category = Category()
new_category.title = category_fields["title"]
new_category.description = category_fields["description"]
new_category.private = category_fields["private"]
new_category.owner = user.id
user.category.append(new_category)
db.session.commit()
return jsonify(category_schema.dump(new_category))
def load_from_sheet(a_sheet):
AssetColl = []
for rowidx in range(a_sheet.nrows):
if a_sheet.cell(rowidx, 1).value == 'Class & Category' or \
a_sheet.cell(rowidx, 1).value == 'Class Code' or \
a_sheet.cell(rowidx, 1).value == 'nvarchar' or \
a_sheet.cell(rowidx, 1).value == '' or \
a_sheet.cell(rowidx, 1).value == 20.0:
continue
iterAsset = Category()
iterAsset.class_code = a_sheet.cell(rowidx, 1).value
iterAsset.description = a_sheet.cell(rowidx, 2).value
iterAsset.category_code = a_sheet.cell(rowidx, 3).value
iterAsset.category_description = a_sheet.cell(rowidx, 4).value
AssetColl.append(iterAsset)
return AssetColl
def seed_db():
from datetime import date
from models.User import User # Importing the User model
from models.Profile import Profile # Importing the Profile model
from models.League import League
from models.Member import Member
from models.Fine import Fine
from models.Point import Point
from models.Category import Category
from models.Sprint import Sprint
from main import bcrypt # Hashing module for the passwords
from faker import Faker # Importing the faker module for fake data
import random # Importing random from the python standard library
import copy
import time
faker = Faker()
users = []
leagues = []
categories = []
sprints = []
points = []
fines = []
for i in range(5):
time.sleep(0.2)
user = User()
user.email = f"test{i+1}@test.com"
user.password = bcrypt.generate_password_hash("123456").decode("utf-8")
db.session.add(user)
users.append(user)
db.session.commit()
for i in range(5):
# time.sleep(0.2)
profile = Profile()
profile.username = f"username{i}"
profile.firstname = f"firstname{i}"
profile.lastname = f"lastname{i}"
profile.user_id = users[i].id
db.session.add(profile)
db.session.commit()
for i in range(5):
# time.sleep(0.2)
new_league = League()
new_league.title = f"League title {i}"
new_league.description = f"A nice league to the power of {i}"
new_league.owner = users[i].id
leagues.append(new_league)
db.session.add(new_league)
db.session.commit()
for i in range(5):
# time.sleep(0.2)
owner = Member()
owner.user_id = leagues[i].owner
owner.league_id = i + 1
owner.active = True
db.session.add(owner)
new_member = Member()
new_member.active = True
new_member.league_id = i + 1
new_member.user_id = random.choice(users).id
while new_member.user_id == owner.user_id:
new_member.user_id = random.choice(users).id
db.session.add(new_member)
db.session.commit()
for i in range(5):
new_sprint = Sprint()
new_sprint.title = f"Sprint title #{i}"
new_sprint.meeting_point = f"The Outback"
new_sprint.creation_time = date.today()
league = leagues[i]
new_sprint.league = league
sprints.append(new_sprint)
db.session.commit()
for i in range(5):
# time.sleep(0.2)
new_category = Category()
new_category.title = f"category title {i}"
new_category.description = f"category description {i}"
if i % 2 == 0:
private = True
else:
private = False
new_category.private = private
new_category.owner = random.choice(users).id
new_category.leagues_categories.append(leagues[i])
categories.append(new_category)
db.session.commit()
for i in range(5):
# time.sleep(0.2)
new_fine = Fine()
new_fine.title = f"Title {i}"
new_fine.description = f"Description {i}"
new_fine.amount = i
if i % 2 == 0:
style = "Award"
else:
style = "Fine"
new_fine.style = style
category = categories[i]
new_fine.category = category
fines.append(new_fine)
db.session.commit()
for i in range(4):
# time.sleep(0.2)
new_point = Point()
new_point.creation_time = date.today()
new_point.fine_id = random.choice(fines).id
sprint = sprints[i]
new_point.sprint = sprint
new_point.giver_id = sprint.league.owner
new_point.receiver_id = sprint.league.members[1].id
db.session.commit()
print("Tables seeded")
