home

courses/python_oop/data_class/dataclass_3.py

@@ -1,4 +1,6 @@
 from dataclasses import dataclass, field
+import random
+import string
 
 @dataclass
 class Person:
@@ -53,16 +55,174 @@ class Person:
     last_name: str
     age: int = field(default=20)
 
+'''
 @dataclass
 class Student(Person):
     course: int = 1
     stipend: int = 1000
+'''
+
+# ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
+# Задание на сортировку
+
+@dataclass(order=True)
+class Athlet:
+    sort_index : float = field(init=False, repr=False)
+    name: str
+    coefficient: float = field(repr=False)
+    scores: list = field(default_factory=list, repr=False)
+    
+    def __post_init__(self):
+        self.sort_index = self.coefficient * sum(self.scores)/len(self.scores)
+        print(self.sort_index)
+
+def test_3():
+    sportsmans = [
+    Athlet('Иван', 1.5, [9.0, 8.0, 7.0]),
+    Athlet('Петр', 1.0, [10.0, 9.0, 8.0]),
+    Athlet('Алексей', 1.2, [8.0, 7.0, 6.0])
+    ]
+
+    print(f"Победитель соревнований: {max(sportsmans)}")
+
+# ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
+
+alphabet = string.ascii_uppercase + string.digits
+
+def generate_guid():
+    guid = ''.join(random.choices(alphabet, k=15))
+    return guid
+
+@dataclass(order=True)
+class Student:
+    sort_index : int = field(init=False, repr=False)
+    first_name: str
+    last_name: str
+    tuition: int = field(default=0, repr=False)
+    guid: str = field(init=False, repr=False)
+    email: str = field(init=False)
+
+    def __post_init__(self):
+        self.sort_index = (self.tuition, self.last_name, self.first_name)
+        self.guid = generate_guid()
+        self.email = self.first_name.lower() + '.' + self.last_name.lower() + '@uni.edu'
+
+def test_4():
+    jane = Student('Jane', 'Lee')
+    julia = Student('Julia', 'Doe')
+    jake = Student('Jake', 'Langdon')
+    joy = Student('Joy', 'Smith')
+    print(*sorted([jane, julia, jake, joy]), sep='\n')
+
+# ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
+
+
+@dataclass
+class Promo:
+    code : str
+    discount : int = field(default=0)
+    goods : list = field(default_factory=list)
+
+@dataclass
+class Product:
+    name: str
+    price: float = field(repr=False)
+
+book = Product('Книга', 100.0)
+usb = Product('Флешка', 50.0)
+pen = Product('Ручка', 10.0)
+
+ACTIVE_PROMO = [
+    Promo('new', 20, [pen]),
+    Promo('all_goods', 30),
+    Promo('sale', 50, [book, usb]),
+]
+
+@dataclass
+class Cart:
+    products: list = field(default_factory=list)
+    discount: int = field(default=0, init=False, repr=False)
+    discount_type: str = field(default='', init=False, repr=False)
+    # product_promo: str = field(default='', init=False, repr=False)
+    products_promo: list = field(default_factory=list, init=False, repr=False)
+
+    def add_product(self, product, quantity=1):
+        for i in range(quantity):
+            self.products.append(product)
+
+    def get_total(self):
+
+        '''
+        if self.product_promo != '':
+            s = sum([x.price if x.name != self.product_promo else 
+            x.price*(1 - self.discount/100) for x in self.products])
+            return s
+        '''
+
+        if self.products_promo != []:
+            s = sum([x.price if x.name not in self.products_promo else 
+            x.price*(1 - self.discount/100) for x in self.products])
+            return s
+
+        s = sum([x.price for x in self.products])
+
+        if self.discount == 0:
+            return s
+        else:
+            return s - s/100*self.discount
+
+    def apply_discount(self, value):
+        if value < 1 or value > 100:
+            raise ValueError('Неправильное значение скидки')
+        self.discount = value
+        self.discount_type = 'disc'
+
+    def apply_promo(self, code):
+        data = {x.code:x.discount for x in ACTIVE_PROMO}
+        foo = {x.code:x.goods for x in ACTIVE_PROMO}
+        if code not in data.keys():
+            print(f'Промокода {code} не существует')
+        else:
+            print(f'Промокод {code} успешно применился')
+            self.discount = data[code]
+            if (foo[code] != []):
+                # self.product_promo = foo[code][0].name
+                for x in foo[code]:
+                    self.products_promo.append(x.name)
+                # print(self.products_promo)
+
+def test_5():
+    pass
+
+def test_6():
+
+    cart = Cart()
+    cart.add_product(book, 10)
+    cart.add_product(pen)
+    cart.add_product(book, 5)
+    cart.add_product(usb, 5)
+    cart.add_product(usb, 15)
+    cart.add_product(pen, 2)
+
+    print(cart.get_total())
+
+    # Применение промокода в 50% на книги и флешки
+    cart.apply_promo('sale')
+    print(cart.get_total())
+
+# ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
+
+
 
 
 
 def main():
     # test_1()
-    test_2()
+    # test_2()
+    # test_3()
+    # test_4()
+    # test_5()
+    test_6()
 
 
 if __name__ == '__main__':

courses/python_oop/enum/enum_1.py

@@ -0,0 +1,25 @@
+from enum import Enum
+
+class Direction(Enum):
+    NORTH = 'N'
+    SOUTH = 'S'
+    EAST = 'E'
+    WEST = 'W'
+
+class Size(Enum):
+    S = 'small'
+    M = 'medium'
+    L = 'large'
+    XL = 'extra large'
+    XXL = 'extra extra large'
+
+
+def test_1():
+    print(Direction.NORTH.name)
+    print(Direction.NORTH.value)
+
+def main():
+    test_1()
+
+if __name__ == '__main__':
+    main()