-
Notifications
You must be signed in to change notification settings - Fork 1
/
models.py
93 lines (69 loc) · 2.78 KB
/
models.py
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
# -*- coding: utf-8 -*-
from main import db
from sqlalchemy.orm import relationship
tags_links = db.Table('tags_links', db.metadata,
db.Column('id_tags', db.Integer,
db.ForeignKey('tags.id')),
db.Column('id_torrent', db.Integer,
db.ForeignKey('torrents.id'))
)
class Tags(db.Model):
id = db.Column(db.Integer, primary_key=True, autoincrement=True)
name = db.Column(db.String(255), unique=True)
def __init__(self, name):
self.name = name
class Accounts(db.Model):
id = db.Column(db.Integer, primary_key=True, autoincrement=True)
name = db.Column(db.String(255), unique=True)
password = db.Column(db.String(255))
status = db.Column(db.Integer)
def __init__(self, name, password, status):
self.name = name
self.password = password
self.status = status
class Torrents(db.Model):
id = db.Column(db.Integer, primary_key=True, autoincrement=True)
name = db.Column(db.String)
description = db.Column(db.String)
filename = db.Column(db.String(255))
id_acc = db.Column(db.Integer, db.ForeignKey('accounts.id'))
size = db.Column(db.BigInteger)
tags_ = relationship("Tags", secondary=tags_links)
def __init__(self, name, description, filename, id_acc, size):
self.name = name
self.description = description
self.filename = filename
self.id_acc = id_acc
self.size = size
@staticmethod
def _find_tag(id_torrent):
tags = Tags.query.filter_by(id=id_torrent)
tag = tags.first()
return tag
def _get_tags(self):
return [x.id for x in self.tags_]
def _set_tags(self, value):
for id_torrent in value:
self.tags_.append(self._find_tag(id_torrent))
str_tags = property(_get_tags, _set_tags)
class TorrentsData(db.Model):
id_torrent = db.Column(db.Integer, db.ForeignKey('torrents.id'),
primary_key=True, autoincrement=False)
payload = db.Column(db.String)
def __init__(self, id_torrent, payload):
self.id_torrent = id_torrent
self.payload = payload
class TorrentsFiles(db.Model):
id = db.Column(db.Integer, primary_key=True, autoincrement=True)
id_torrent = db.Column(db.Integer, db.ForeignKey('torrents.id'),
primary_key=False, autoincrement=False)
filename = db.Column(db.String)
def __init__(self, id_torrent, filename):
self.id_torrent = id_torrent
self.filename = filename
class TorrentsStat(db.Model):
id = db.Column(db.Integer, primary_key=True, autoincrement=False)
size = db.Column(db.BigInteger)
def __init__(self, id, size):
self.id = id
self.size = size