-
Notifications
You must be signed in to change notification settings - Fork 0
/
server.py
67 lines (49 loc) · 1.36 KB
/
server.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
import os
from sanic import Sanic
from sanic.views import HTTPMethodView
from sanic.response import html, redirect, text, json
import pony.orm as P
DB_dir = os.path.abspath(os.path.dirname(__file__))
DB_uri = os.path.join(DB_dir, 'database.db')
DB = P.Database()
DB.bind(
'postgres',
user='postgres', password='pgsql',
host='localhost', database='books'
)
class Us(DB.Entity):
name = P.Required(str)
lastname = P.Required(str)
uname = P.PrimaryKey(str, 32) # id
email = P.Required(str, 60, unique=True)
pwd = P.Required(str)
books = P.Set('Book')
class Book(DB.Entity):
book_id = P.PrimaryKey(str)
users = P.Set(Us)
P.sql_debug(True)
DB.generate_mapping(create_tables=True)
class Server():
"""docstring for Server"""
def __init__(self):
pass
@P.db_session
def add_us(self, **kw):
try:
new = Us(**kw)
P.flush()
return new
except Exception as e:
raise e
def auth_user(self, uname, pwd):
in_pwd = DB.select(u.pwd for u in Us if u.uname == uname)
return in_pwd is pwd
@P.db_session
def us_exist(self, uname=None, email=None):
return bool(
P.select(
u for u in Us if u.uname == uname or u.email == email
).first()
)
def __del__(self):
DB.disconnect()