Fix pylint for models files
This commit is contained in:
parent
d21f2d48f5
commit
ec82639c5c
8 changed files with 43 additions and 17 deletions
|
@ -1,3 +1,5 @@
|
||||||
|
"The base file for everything related to the models"
|
||||||
|
# pylint: disable=C0301
|
||||||
# This file will expose what we want from the models module
|
# This file will expose what we want from the models module
|
||||||
# This will probably be everything. But putting the imports here makes it possible to import all models in one line like this:
|
# This will probably be everything. But putting the imports here makes it possible to import all models in one line like this:
|
||||||
#
|
#
|
||||||
|
|
|
@ -1,3 +1,7 @@
|
||||||
|
"AnonymouseUser for people who are not logged in the normal way"
|
||||||
|
# pylint: disable=R0201,C0111
|
||||||
|
|
||||||
|
|
||||||
class AnonymouseUser:
|
class AnonymouseUser:
|
||||||
id = None
|
id = None
|
||||||
|
|
||||||
|
|
|
@ -1,3 +1,4 @@
|
||||||
|
# pylint: disable=C0111
|
||||||
from flask_sqlalchemy import SQLAlchemy
|
from flask_sqlalchemy import SQLAlchemy
|
||||||
|
|
||||||
db = SQLAlchemy()
|
db = SQLAlchemy()
|
||||||
|
|
|
@ -1,9 +1,11 @@
|
||||||
|
"Script for everything Location related in the database"
|
||||||
import typing
|
import typing
|
||||||
|
|
||||||
from models import db
|
from models import db
|
||||||
|
|
||||||
|
|
||||||
class Location(db.Model):
|
class Location(db.Model):
|
||||||
|
"Class used for configuring the Location model in the database"
|
||||||
id = db.Column(db.Integer, primary_key=True)
|
id = db.Column(db.Integer, primary_key=True)
|
||||||
name = db.Column(db.String(120), nullable=False)
|
name = db.Column(db.String(120), nullable=False)
|
||||||
address = db.Column(db.String(254))
|
address = db.Column(db.String(254))
|
||||||
|
@ -12,9 +14,9 @@ class Location(db.Model):
|
||||||
products = db.relationship("Product", backref="location", lazy="dynamic")
|
products = db.relationship("Product", backref="location", lazy="dynamic")
|
||||||
orders = db.relationship("Order", backref="location", lazy="dynamic")
|
orders = db.relationship("Order", backref="location", lazy="dynamic")
|
||||||
|
|
||||||
def configure(
|
def configure(self, name: str, address: str,
|
||||||
self, name: str, address: str, telephone: typing.Optional[str], website: str
|
telephone: typing.Optional[str], website: str) -> None:
|
||||||
) -> None:
|
"Configure the Location"
|
||||||
self.name = name
|
self.name = name
|
||||||
self.address = address
|
self.address = address
|
||||||
self.website = website
|
self.website = website
|
||||||
|
|
|
@ -1,3 +1,4 @@
|
||||||
|
"Script for everything Order related in the database"
|
||||||
import typing
|
import typing
|
||||||
from datetime import datetime
|
from datetime import datetime
|
||||||
|
|
||||||
|
@ -7,6 +8,7 @@ from .user import User
|
||||||
|
|
||||||
|
|
||||||
class Order(db.Model):
|
class Order(db.Model):
|
||||||
|
"Class used for configuring the Order model in the database"
|
||||||
id = db.Column(db.Integer, primary_key=True)
|
id = db.Column(db.Integer, primary_key=True)
|
||||||
courrier_id = db.Column(db.Integer, nullable=True)
|
courrier_id = db.Column(db.Integer, nullable=True)
|
||||||
location_id = db.Column(db.Integer, db.ForeignKey("location.id"))
|
location_id = db.Column(db.Integer, db.ForeignKey("location.id"))
|
||||||
|
@ -15,31 +17,31 @@ class Order(db.Model):
|
||||||
public = db.Column(db.Boolean, default=True)
|
public = db.Column(db.Boolean, default=True)
|
||||||
items = db.relationship("OrderItem", backref="order", lazy="dynamic")
|
items = db.relationship("OrderItem", backref="order", lazy="dynamic")
|
||||||
|
|
||||||
def configure(
|
def configure(self, courrier: User, location: Location,
|
||||||
self,
|
starttime: db.DateTime, stoptime: db.DateTime,) -> None:
|
||||||
courrier: User,
|
"Configure the Order"
|
||||||
location: Location,
|
# pylint: disable=W0201
|
||||||
starttime: db.DateTime,
|
|
||||||
stoptime: db.DateTime,
|
|
||||||
) -> None:
|
|
||||||
self.courrier = courrier
|
self.courrier = courrier
|
||||||
self.location = location
|
self.location = location
|
||||||
self.starttime = starttime
|
self.starttime = starttime
|
||||||
self.stoptime = stoptime
|
self.stoptime = stoptime
|
||||||
|
|
||||||
def __repr__(self) -> str:
|
def __repr__(self) -> str:
|
||||||
|
# pylint: disable=R1705
|
||||||
if self.location:
|
if self.location:
|
||||||
return "Order %d @ %s" % (self.id, self.location.name or "None")
|
return "Order %d @ %s" % (self.id, self.location.name or "None")
|
||||||
else:
|
else:
|
||||||
return "Order %d" % (self.id)
|
return "Order %d" % (self.id)
|
||||||
|
|
||||||
def group_by_user(self) -> typing.Dict[str, typing.Any]:
|
def group_by_user(self) -> typing.Dict[str, typing.Any]:
|
||||||
|
"Group items of an Order by user"
|
||||||
group: typing.Dict[str, typing.Any] = dict()
|
group: typing.Dict[str, typing.Any] = dict()
|
||||||
for item in self.items:
|
for item in self.items:
|
||||||
user = group.get(item.get_name(), dict())
|
user = group.get(item.get_name(), dict())
|
||||||
user["total"] = user.get("total", 0) + item.product.price
|
user["total"] = user.get("total", 0) + item.product.price
|
||||||
user["to_pay"] = (
|
user["to_pay"] = (
|
||||||
user.get("to_pay", 0) + item.product.price if not item.paid else 0
|
user.get("to_pay", 0) +
|
||||||
|
item.product.price if not item.paid else 0
|
||||||
)
|
)
|
||||||
user["paid"] = user.get("paid", True) and item.paid
|
user["paid"] = user.get("paid", True) and item.paid
|
||||||
user["products"] = user.get("products", []) + [item.product]
|
user["products"] = user.get("products", []) + [item.product]
|
||||||
|
@ -48,6 +50,7 @@ class Order(db.Model):
|
||||||
return group
|
return group
|
||||||
|
|
||||||
def group_by_product(self) -> typing.Dict[str, typing.Any]:
|
def group_by_product(self) -> typing.Dict[str, typing.Any]:
|
||||||
|
"Group items of an Order by product"
|
||||||
group: typing.Dict[str, typing.Any] = dict()
|
group: typing.Dict[str, typing.Any] = dict()
|
||||||
for item in self.items:
|
for item in self.items:
|
||||||
product = group.get(item.product.name, dict())
|
product = group.get(item.product.name, dict())
|
||||||
|
@ -59,6 +62,7 @@ class Order(db.Model):
|
||||||
return group
|
return group
|
||||||
|
|
||||||
def can_close(self, user_id: int) -> bool:
|
def can_close(self, user_id: int) -> bool:
|
||||||
|
"Check if a user can close the Order"
|
||||||
if self.stoptime and self.stoptime < datetime.now():
|
if self.stoptime and self.stoptime < datetime.now():
|
||||||
return False
|
return False
|
||||||
user = None
|
user = None
|
||||||
|
|
|
@ -1,3 +1,4 @@
|
||||||
|
"Script for everything OrderItem related in the database"
|
||||||
from datetime import datetime
|
from datetime import datetime
|
||||||
|
|
||||||
from .database import db
|
from .database import db
|
||||||
|
@ -7,6 +8,7 @@ from .user import User
|
||||||
|
|
||||||
|
|
||||||
class OrderItem(db.Model):
|
class OrderItem(db.Model):
|
||||||
|
"Class used for configuring the OrderItem model in the database"
|
||||||
id = db.Column(db.Integer, primary_key=True)
|
id = db.Column(db.Integer, primary_key=True)
|
||||||
user_id = db.Column(db.Integer, db.ForeignKey("user.id"))
|
user_id = db.Column(db.Integer, db.ForeignKey("user.id"))
|
||||||
order_id = db.Column(db.Integer, db.ForeignKey("order.id"), nullable=False)
|
order_id = db.Column(db.Integer, db.ForeignKey("order.id"), nullable=False)
|
||||||
|
@ -20,11 +22,14 @@ class OrderItem(db.Model):
|
||||||
name = db.Column(db.String(120))
|
name = db.Column(db.String(120))
|
||||||
|
|
||||||
def configure(self, user: User, order: Order, product: Product) -> None:
|
def configure(self, user: User, order: Order, product: Product) -> None:
|
||||||
|
"Configure the OrderItem"
|
||||||
|
# pylint: disable=W0201
|
||||||
self.user = user
|
self.user = user
|
||||||
self.order = order
|
self.order = order
|
||||||
self.product = product
|
self.product = product
|
||||||
|
|
||||||
def get_name(self) -> str:
|
def get_name(self) -> str:
|
||||||
|
"Get the name of the user which 'owns' the item"
|
||||||
if self.user_id is not None and self.user_id > 0:
|
if self.user_id is not None and self.user_id > 0:
|
||||||
return self.user.username
|
return self.user.username
|
||||||
return self.name
|
return self.name
|
||||||
|
@ -39,7 +44,9 @@ class OrderItem(db.Model):
|
||||||
product_name or "None",
|
product_name or "None",
|
||||||
)
|
)
|
||||||
|
|
||||||
|
# pylint: disable=W0613
|
||||||
def can_delete(self, order_id: int, user_id: int, name: str) -> bool:
|
def can_delete(self, order_id: int, user_id: int, name: str) -> bool:
|
||||||
|
"Check if a user can delete an item"
|
||||||
if int(self.order_id) != int(order_id):
|
if int(self.order_id) != int(order_id):
|
||||||
return False
|
return False
|
||||||
if self.order.stoptime and self.order.stoptime < datetime.now():
|
if self.order.stoptime and self.order.stoptime < datetime.now():
|
||||||
|
|
|
@ -1,23 +1,25 @@
|
||||||
|
"Script for everything Product related in the database"
|
||||||
from models import db
|
from models import db
|
||||||
|
|
||||||
from .location import Location
|
from .location import Location
|
||||||
|
|
||||||
|
|
||||||
class Product(db.Model):
|
class Product(db.Model):
|
||||||
|
"Class used for configuring the Product model in the database"
|
||||||
id = db.Column(db.Integer, primary_key=True)
|
id = db.Column(db.Integer, primary_key=True)
|
||||||
location_id = db.Column(db.Integer, db.ForeignKey("location.id"))
|
location_id = db.Column(db.Integer, db.ForeignKey("location.id"))
|
||||||
name = db.Column(db.String(120), nullable=False)
|
name = db.Column(db.String(120), nullable=False)
|
||||||
price = db.Column(db.Integer, nullable=False)
|
price = db.Column(db.Integer, nullable=False)
|
||||||
orderItems = db.relationship("OrderItem", backref="product", lazy="dynamic")
|
orderItems = db.relationship("OrderItem",
|
||||||
|
backref="product", lazy="dynamic")
|
||||||
|
|
||||||
def configure(self, location: Location, name: str, price: int) -> None:
|
def configure(self, location: Location, name: str, price: int) -> None:
|
||||||
|
"Configure the Product"
|
||||||
|
# pylint: disable=W0201
|
||||||
self.location = location
|
self.location = location
|
||||||
self.name = name
|
self.name = name
|
||||||
self.price = price
|
self.price = price
|
||||||
|
|
||||||
def __repr__(self) -> str:
|
def __repr__(self) -> str:
|
||||||
return "%s (€%d)from %s" % (
|
return "%s (€%d)from %s" % (self.name, self.price / 100,
|
||||||
self.name,
|
self.location or "None",)
|
||||||
self.price / 100,
|
|
||||||
self.location or "None",
|
|
||||||
)
|
|
||||||
|
|
|
@ -1,7 +1,9 @@
|
||||||
|
"Script for everything User related in the database"
|
||||||
from models import db
|
from models import db
|
||||||
|
|
||||||
|
|
||||||
class User(db.Model):
|
class User(db.Model):
|
||||||
|
"Class used for configuring the User model in the database"
|
||||||
id = db.Column(db.Integer, primary_key=True)
|
id = db.Column(db.Integer, primary_key=True)
|
||||||
username = db.Column(db.String(80), unique=True, nullable=False)
|
username = db.Column(db.String(80), unique=True, nullable=False)
|
||||||
admin = db.Column(db.Boolean)
|
admin = db.Column(db.Boolean)
|
||||||
|
@ -15,10 +17,12 @@ class User(db.Model):
|
||||||
orderItems = db.relationship("OrderItem", backref="user", lazy="dynamic")
|
orderItems = db.relationship("OrderItem", backref="user", lazy="dynamic")
|
||||||
|
|
||||||
def configure(self, username: str, admin: bool, bias: int) -> None:
|
def configure(self, username: str, admin: bool, bias: int) -> None:
|
||||||
|
"Configure the User"
|
||||||
self.username = username
|
self.username = username
|
||||||
self.admin = admin
|
self.admin = admin
|
||||||
self.bias = bias
|
self.bias = bias
|
||||||
|
|
||||||
|
# pylint: disable=C0111, R0201
|
||||||
def is_authenticated(self) -> bool:
|
def is_authenticated(self) -> bool:
|
||||||
return True
|
return True
|
||||||
|
|
||||||
|
|
Loading…
Reference in a new issue