forked from kompetenzinventar/ki-backend
84 lines
2.9 KiB
Python
84 lines
2.9 KiB
Python
|
import csv
|
||
|
|
import logging
|
||
|
|
|
||
|
|
from app import app, db
|
||
|
|
from ki.models import Address, Contact, ContactType, Language, Skill, Profile, ProfileLanguage, ProfileSkill, User
|
||
|
|
|
||
|
|
|
||
|
|
def seed(dev: bool):
|
||
|
|
skill_seed_file_path = app.config["KI_DATA_DIR"] + "/seed_data/skills.csv"
|
||
|
|
|
||
|
|
logging.info("importing skills")
|
||
|
|
|
||
|
|
with open(skill_seed_file_path) as skills_file:
|
||
|
|
skills_csv_reader = csv.DictReader(skills_file)
|
||
|
|
|
||
|
|
for skill in skills_csv_reader:
|
||
|
|
id = int(skill["id"])
|
||
|
|
db_skill = Skill.query.get(id)
|
||
|
|
|
||
|
|
if db_skill is None:
|
||
|
|
db.session.add(Skill(id=int(skill["id"]), name=skill["name"]))
|
||
|
|
|
||
|
|
logging.info("importing languages")
|
||
|
|
|
||
|
|
iso_seed_file_path = app.config["KI_DATA_DIR"] + "/seed_data/iso_639_1.csv"
|
||
|
|
|
||
|
|
with open(iso_seed_file_path) as iso_file:
|
||
|
|
iso_csv_reader = csv.DictReader(iso_file)
|
||
|
|
|
||
|
|
for iso in iso_csv_reader:
|
||
|
|
id = iso["639-1"]
|
||
|
|
db_language = Language.query.get(id)
|
||
|
|
|
||
|
|
if db_language is None:
|
||
|
|
db.session.add(Language(id=iso["639-1"], name=iso["Sprache"]))
|
||
|
|
|
||
|
|
if dev:
|
||
|
|
logging.info("seeding peter :)")
|
||
|
|
|
||
|
|
peter = User(auth_id="peter")
|
||
|
|
db.session.add(peter)
|
||
|
|
|
||
|
|
peters_profile = Profile(nickname="peternichtlustig",
|
||
|
|
pronouns="Herr Dr. Dr.",
|
||
|
|
volunteerwork="Gartenverein",
|
||
|
|
freetext="Ich mag Kaffee",
|
||
|
|
user=peter)
|
||
|
|
db.session.add(peters_profile)
|
||
|
|
|
||
|
|
matrix_type = ContactType(name="Matrix")
|
||
|
|
db.session.add(matrix_type)
|
||
|
|
|
||
|
|
matrix_contact = Contact(profile=peters_profile,
|
||
|
|
contacttype=matrix_type,
|
||
|
|
content="@peter:wtf-eg.de")
|
||
|
|
db.session.add(matrix_contact)
|
||
|
|
|
||
|
|
peters_address = Address(name="Peter Nichtlustig",
|
||
|
|
street="Waldweg",
|
||
|
|
house_number="23i",
|
||
|
|
additional="Hinterhaus",
|
||
|
|
postcode="13337",
|
||
|
|
city="Bielefeld",
|
||
|
|
country="Deutschland",
|
||
|
|
profile=peters_profile)
|
||
|
|
db.session.add(peters_address)
|
||
|
|
|
||
|
|
peters_python_skill = ProfileSkill(profile=peters_profile,
|
||
|
|
skill_id=3,
|
||
|
|
level=5)
|
||
|
|
db.session.add(peters_python_skill)
|
||
|
|
|
||
|
|
peter_de = ProfileLanguage(profile=peters_profile,
|
||
|
|
language_id="de",
|
||
|
|
level=5)
|
||
|
|
db.session.add(peter_de)
|
||
|
|
|
||
|
|
peter_fr = ProfileLanguage(profile=peters_profile,
|
||
|
|
language_id="fr",
|
||
|
|
level=3)
|
||
|
|
db.session.add(peter_fr)
|
||
|
|
|
||
|
|
db.session.commit()
|