MasterSoundBackend/app/master_sound/models.py at 91105b4f3c1f127b1d34f88393b79074bd2459b6 · MasterSoundC5/MasterSoundBackend · 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
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
from datetime import datetime

import requests

from app.db import db, BaseModelMixin


# TODO:
# It's doesn't recommend handle migrations in the same file that the models
# And the migrations should be handle with scripts

albums_artists = db.Table(
        'albums_artists',
        db.Model.metadata,
        db.Column('albums_artists_id', db.Integer, primary_key=True),
        db.Column('artist_id', db.Integer, db.ForeignKey('artists.artist_id', ondelete='CASCADE'), nullable=False),
        db.Column('album_id', db.Integer, db.ForeignKey('albums.album_id', ondelete='CASCADE'), nullable=False)
        )

playlists_songs = db.Table(
        'playlists_songs',
        db.Model.metadata,
        db.Column('playlists_songs_id', db.Integer, primary_key=True),
        db.Column('playlist_id', db.Integer, db.ForeignKey('playlists.playlist_id', ondelete='CASCADE'), nullable=False),
        db.Column('song_id', db.Integer, db.ForeignKey('songs.song_id', ondelete='CASCADE'), nullable=False)
        )

played_songs = db.Table(
        'played_songs',
        db.Column('played_song_id', db.Integer, primary_key=True),
        db.Column('song_id', db.Integer, db.ForeignKey('songs.song_id', ondelete='CASCADE',), nullable=False),
        db.Column('user_id', db.Integer, db.ForeignKey('users.user_id', ondelete='CASCADE',), nullable=False),
        db.Column('created_at', db.DateTime, default=datetime.utcnow),
        db.Column('updated_at', db.DateTime, default=datetime.utcnow, onupdate=datetime.utcnow),
        db.Column('active', db.Integer, default=1)
        )


class Country(db.Model, BaseModelMixin):
    __tablename__ = 'countries'

    country_id = db.Column(db.Integer, primary_key=True)
    iso = db.Column(db.String(2), nullable=False)
    name = db.Column(db.String(45), nullable=False)
    spanish_name = db.Column(db.String(45), nullable=False)
    users = db.relationship('User', back_populates='country', cascade='all, delete, delete-orphan', passive_deletes=True)


class User(db.Model, BaseModelMixin):
    __tablename__ = 'users'

    user_id = db.Column(db.Integer, primary_key=True)
    given_name = db.Column(db.String(30), nullable=False)
    last_name = db.Column(db.String(30), nullable=False)
    email = db.Column(db.String(50), nullable=False, unique=True)
    password = db.Column(db.String(60), nullable=False)
    country_id = db.Column(db.Integer, db.ForeignKey('countries.country_id', ondelete='CASCADE'), nullable=False)
    image_url = db.Column(db.String(100), nullable=False)
    sex = db.Column(db.String(1), nullable=False)
    created_at = db.Column(db.DateTime, default=datetime.utcnow)
    updated_at = db.Column(db.DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)
    active = db.Column(db.Integer, nullable=False, default=1)
    country = db.relationship('Country', back_populates='users')
    played_songs = db.relationship('Song', back_populates='playing_users', secondary=played_songs, passive_deletes=True)
    playlists = db.relationship('Playlist', back_populates='user')


class Album(db.Model, BaseModelMixin):
    __tablename__ = 'albums'

    album_id = db.Column(db.Integer, primary_key=True)
    cover_image_url = db.Column(db.String(100))
    spt_album_id = db.Column(db.String(30), nullable=False, unique=True)
    album_name = db.Column(db.String(80), nullable=False)
    songs = db.relationship('Song', back_populates='album', cascade='all, delete, delete-orphan', passive_deletes=True)
    created_at = db.Column(db.DateTime, default=datetime.utcnow)
    updated_at = db.Column(db.DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)
    active = db.Column(db.Integer, default=1)
    artists = db.relationship('Artist', back_populates='albums', secondary=albums_artists, cascade='all, delete')


class Artist(db.Model, BaseModelMixin):
    __tablename__ = 'artists'

    artist_id = db.Column(db.Integer, primary_key=True)
    spt_artist_id = db.Column(db.String(30), nullable=False, unique=True)
    artist_name = db.Column(db.String(50), nullable=False)
    cover_image_url = db.Column(db.String(100))
    created_at = db.Column(db.DateTime, default=datetime.utcnow)
    updated_at = db.Column(db.DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)
    active = db.Column(db.Integer, default=1)
    albums = db.relationship('Album', back_populates='artists', secondary=albums_artists, passive_deletes=True)


class Song(db.Model, BaseModelMixin):
    __tablename__ = 'songs'

    song_id = db.Column(db.Integer, primary_key=True)
    spt_song_id = db.Column(db.String(30), nullable=False, unique=True)
    song_name = db.Column(db.String(120), nullable=False)
    album_id = db.Column(db.Integer, db.ForeignKey('albums.album_id', ondelete='CASCADE'), nullable=False)
    order_number = db.Column(db.Integer, nullable=False)
    duration = db.Column(db.String(6), nullable=False)
    sound_url = db.Column(db.String(200), nullable=False)
    created_at = db.Column(db.DateTime, default=datetime.utcnow)
    updated_at = db.Column(db.DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)
    active = db.Column(db.Integer, default=1)
    album = db.relationship('Album', back_populates='songs')
    playlists = db.relationship('Playlist', back_populates='songs', secondary=playlists_songs, cascade='all, delete')
    playing_users = db.relationship('User', back_populates='played_songs', secondary=played_songs, cascade='all, delete')


class Playlist(db.Model, BaseModelMixin):
    __tablename__ = 'playlists'

    playlist_id = db.Column(db.Integer, primary_key=True)
    user_id = db.Column(db.Integer, db.ForeignKey('users.user_id', ondelete='CASCADE'), nullable=False)
    playlist_name = db.Column(db.String(50), nullable=False)
    favourite = db.Column(db.Integer, default=0)
    created_at = db.Column(db.DateTime, default=datetime.utcnow)
    updated_at = db.Column(db.DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)
    active = db.Column(db.Integer, default=1)
    user = db.relationship('User', back_populates='playlists')
    songs = db.relationship('Song', back_populates='playlists', secondary=playlists_songs, passive_deletes=True)