-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathmodels.py
More file actions
55 lines (44 loc) · 1.66 KB
/
models.py
File metadata and controls
55 lines (44 loc) · 1.66 KB
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
from flask import Flask, json
from settings import app
from flask_sqlalchemy import SQLAlchemy
db = SQLAlchemy(app)
class Book(db.Model):
__tablename__ = 'books'
id = db.Column(db.Integer, primary_key=True)
name = db.Column(db.String(80), nullable=False)
price = db.Column(db.Float, nullable=False)
isbn = db.Column(db.String(13), nullable=False, unique=True)
def json(self):
return {"name": self.name, "price": self.price, "isbn": self.isbn}
def add_new(_name, _price, _isbn):
new_book = Book(name=_name, price=_price, isbn=_isbn)
db.session.add(new_book)
db.session.commit()
def get_all():
return [Book.json(book) for book in Book.query.all()]
def get_by(_isbn):
return Book.query.filter_by(isbn=_isbn).first()
def delete_by(_isbn):
Book.query.filter_by(isbn=_isbn).delete()
db.session.commit()
def change_by(_isbn, **kwargs):
replaced_object = Book.query.filter_by(isbn=_isbn).first()
if "price" in kwargs.keys():
replaced_object.price = kwargs["price"]
if "name" in kwargs.keys():
replaced_object.name = kwargs["name"]
db.session.commit()
def replace_by(_isbn, _name, _price):
replaced_object = Book.query.filter_by(isbn=_isbn).first()
print(type(replaced_object))
replaced_object.price = _price
replaced_object.name = _name
db.session.commit()
def __repr__(self):
book_object = {
"name": self.name,
"price": self.price,
"isbn": self.isbn
}
return json.dumps(book_object)
# used only for console info