jonny/matemat
1
0
Fork 0
forked from s3lph/matemat
Code Pull requests Activity

Reworked Product to allow a 0 argument constructor, because unittest apparently breaks stuff...

Browse source
This commit is contained in:
s3lph 2018-05-29 23:36:52 +02:00
parent 1439c3072c
commit 3794a570c8
2 changed files with 27 additions and 23 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
matemat/db/database.py
View file

@ -254,7 +254,7 @@ class Database(object):
FROM products
'''):
product_id, name, price_member, price_external = row
products.append(Product(product_id, name, price_member, price_external))
products.append(Product.from_database(product_id, name, price_member, price_external))
return products
def create_product(self, name: str, price_member: int, price_non_member: int) -> Product:
@ -273,9 +273,11 @@ class Database(object):
})
c.execute('SELECT last_insert_rowid()')
product_id = int(c.fetchone()[0])
return Product(product_id, name, price_member, price_non_member)
return Product.from_database(product_id, name, price_member, price_non_member)
def change_product(self, product: Product):
if product.id == -1:
raise ValueError('Invalid product ID')
with self.transaction() as c:
c.execute('''
UPDATE products
@ -291,6 +293,8 @@ class Database(object):
})
def delete_product(self, product: Product):
if product.id == -1:
raise ValueError('Invalid product ID')
with self.transaction() as c:
c.execute('''
DELETE FROM products
@ -298,6 +302,8 @@ class Database(object):
''', [product.id])
def increment_consumption(self, user: User, product: Product, count: int = 1):
if product.id == -1:
raise ValueError('Invalid product ID')
with self.transaction() as c:
c.execute('''
SELECT count
@ -342,6 +348,8 @@ class Database(object):
})
def restock(self, product: Product, count: int):
if product.id == -1:
raise ValueError('Invalid product ID')
with self.transaction() as c:
c.execute('''
UPDATE products

38
matemat/primitives/Product.py
View file

@ -1,28 +1,24 @@
from inspect import stack
class Product(object):
def __init__(self):
print(stack())
self._product_id: int = 0
self._name: str = ''
self._price_member: int = 0
self._price_non_member: int = 0
raise NotImplementedError('This shoudt not be called!')
def __init__(self,
product_id: int = -1,
name: str = '',
price_member: int = 0,
price_non_member: int = 0):
self._product_id: int = product_id
self._name: str = name
self._price_member: int = price_member
self._price_non_member: int = price_non_member
# def __init__(self,
# product_id: int,
# name: str,
# price_member: int,
# price_non_member: int):
# if product_id == -1:
# raise ValueError('Invalid product ID')
# self._product_id: int = product_id
# self._name: str = name
# self._price_member: int = price_member
# self._price_non_member: int = price_non_member
@classmethod
def from_database(cls,
product_id: int,
name: str,
price_member: int,
price_non_member: int) -> 'Product':
product = cls(product_id, name, price_member, price_non_member)
return product
@property
def id(self) -> int: