seed.py

import datetime
from sqlalchemy import func

from model import connect_to_db, db, Character, Relationship, Series, User, CharacterSeries, CharacterRating, SeriesRating

from server import app

import bcrypt


def load_characters():

    for line in open("csv/characters.csv"):
        line = line[:-1]
        name, group, actor, bio, image = line.split(',')

        character = Character(name=name,
                              group=group,
                              actor=actor,
                              bio=bio,
                              image=image)

        db.session.add(character)

    db.session.commit()


def load_series():

    for line in open("csv/series.csv"):
        line = line[:-1]
        name, synopsis, date, chron_order, image = line.split(',')

        series = Series(name=name,
                        synopsis=synopsis,
                        date=date,
                        chron_order=chron_order,
                        image=image)

        db.session.add(series)

    db.session.commit()


def load_relationships():

    for line in open("csv/relationships.csv"):
        line = line[:-1]
        char1_name, char2_name, threshold = line.split(',')

        character_1 = Character.query.filter_by(name=char1_name).first()
        character_2 = Character.query.filter_by(name=char2_name).first()

        relationship = Relationship(char1_id=character_1.char_id,
                                    char2_id=character_2.char_id,
                                    threshold=threshold)

        db.session.add(relationship)

    db.session.commit()


def load_appearances():

    for line in open("csv/characterseries.csv"):
        line = line[:-1]
        char_name, series_name = line.split(',')

        character = Character.query.filter_by(name=char_name).first()
        series = Series.query.filter_by(name=series_name).first()

        appearance = CharacterSeries(character=character.char_id,
                                     series=series.series_id)

        db.session.add(appearance)

    db.session.commit()


def load_users():

    for line in open("csv/users.csv"):
        line = line[:-1]
        email, password = line.split(',')
        password = password.encode('utf8')
        hashed = bcrypt.hashpw(password, bcrypt.gensalt())

        user = User(email=email,
                    password=hashed)


        db.session.add(user)

    db.session.commit()


def load_character_ratings():

    for line in open("csv/char_ratings.csv"):
        line = line[:-1]
        char_id, user_id, score = line.split(',')

        char_id = int(char_id)
        user_id = int(user_id)
        score = int(score)

        rating = CharacterRating(char_id=char_id,
                        user_id=user_id,
                        score=score)

        db.session.add(rating)

    db.session.commit()


def load_series_ratings():

    for line in open("csv/series_ratings.csv"):
        line = line[:-1]
        series_id, user_id, score = line.split(',')

        series_id = int(char_id)
        user_id = int(user_id)
        score = int(score)

        rating = SeriesRating(series_id=series_id,
                        user_id=user_id,
                        score=score)

        db.session.add(rating)

    db.session.commit()


if __name__ == "__main__":
    connect_to_db(app)
    db.drop_all()
    db.create_all()
    load_characters()
    load_series()
    load_relationships()
    load_appearances()
    load_users()
    load_character_ratings()
    load_series_ratings()

    # Run the functions (loading data, setting value of ids)