Pokedex/db_setup.py at develop · defiled/Pokedex · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
from sqlalchemy import Column, ForeignKey, Integer, String, Text
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import relationship
from sqlalchemy import create_engine

Base = declarative_base()


class User(Base):
    __tablename__ = 'user'
    id = Column(Integer, primary_key=True)
    name = Column(String(250), nullable=False)
    password = Column(String(250))
    email = Column(String(250), nullable=False)
    picture = Column(String(250))

    pokemon = relationship("UserPokemon")


class Pokemon(Base):
    __tablename__ = 'pokemon'
    id = Column(Integer, primary_key=True)
    name = Column(String(250))
    description = Column(Text)
    weight = Column(Integer)
    height = Column(Integer)
    generation = Column(String(250))
    region_name = Column(String(250))

    pokemon_sprite = relationship("PokemonSprites", uselist=False,
                                  back_populates="pokemon")
    user_pokemon = relationship("UserPokemon")
    pokemon_type = relationship("PokemonTypes", back_populates="pokemon")

    @property
    def serialize(self):
        return {
            'id': self.id,
            'name': self.name,
            'description': self.description,
            'weight': self.weight,
            'height': self.height,
            'generation': self.generation,
            'region_name': self.region_name,
        }


class UserPokemon(Base):
    __tablename__ = 'user_pokemon'
    id = Column(Integer, primary_key=True)
    pokemon_id = Column(Integer, ForeignKey('pokemon.id'))
    user_id = Column(Integer, ForeignKey('user.id'))
    nickname = Column(String(250))
    level = Column(String(250))
    party_order = Column(Integer)

    pokemon = relationship("Pokemon")
    user = relationship("User")


class Types(Base):
    __tablename__ = 'types'
    id = Column(Integer, primary_key=True)
    name = Column(String(250), nullable=False)

    pokemon_type = relationship("PokemonTypes", back_populates="type")

    @property
    def serialize(self):
        return {
            'id': self.id,
            'name': self.name,
        }


class PokemonTypes(Base):
    __tablename__ = 'pokemon_types'
    id = Column(Integer, primary_key=True)
    pokemon_id = Column(Integer, ForeignKey('pokemon.id'))
    type_id = Column(Integer, ForeignKey('types.id'))

    type = relationship("Types", back_populates="pokemon_type")
    pokemon = relationship('Pokemon', back_populates='pokemon_type')


class PokemonSprites(Base):
    __tablename__ = 'pokemon_sprites'
    id = Column(Integer, primary_key=True)
    name = Column(String(250), nullable=False)
    sprite_url = Column(String(250), nullable=False)
    pokemon_id = Column(Integer, ForeignKey('pokemon.id'))

    pokemon = relationship("Pokemon", back_populates="pokemon_sprite")

    @property
    def serialize(self):
        return {
            'id': self.id,
            'name': self.name,
            'sprite_url': self.sprite_url,
            'pokemon_id': self.pokemon_id
        }


engine = create_engine('postgresql://catalog:catalog@localhost/pokedex')
Base.metadata.create_all(engine)