Flyweight/python/Flyweight.py - master - telnov.ob/patterns

Flyweight/python/Flyweight.py
105 строк3 KB
georgedem
Flyweight
22 авг 2022, 00:23
22 авг 2022, 00:23f623b2b
from typing import List
from typing import Dict


class Shared:

    def __init__(self, company: str, position: str):
        self.__company = company
        self.__position = position

    @property
    def company(self):
        return self.__company

    @property
    def position(self):
        return self.__position


class Unique:

    def __init__(self, name: str, passport: str):
        self.__name = name
        self.__passport = passport

    @property
    def name(self):
        return self.__name

    @property
    def passport(self):
        return self.__passport


class Flyweight:

    def __init__(self, shared: Shared):
        self.__shared = shared

    def process(self, unique: Unique):
        print("New data: shared - {}_{} unique - {}_{}".format(self.__shared.company, self.__shared.position, unique.name, unique.passport))

    def get_data(self) -> str:
        return self.__shared.company + '_' + self.__shared.position


class FlyweightFactory:

    def get_key(self, shared: Shared):
        return '{}_{}'.format(shared.company, shared.position)

    def __init__(self, shareds: List[Shared]):
        self.__flyweights: Dict[str, Flyweight] = {}
        for shared in shareds:
            self.__flyweights[self.get_key(shared)] = Flyweight(shared)

    def get_flyweight(self, shared: Shared):
        key: str = self.get_key(shared)
        if self.__flyweights.get(key) is None:
            print("Flyweight factory: object not found " + key)
            self.__flyweights[key] = Flyweight(shared)
        else:
            print("Flyweight factory: object found " + key)
        return self.__flyweights[key]

    def list_flyweights(self):
        count: int = len(self.__flyweights)
        print("\nFlyweight factory: count = {}".format(count))
        for pair in self.__flyweights.values():
            print(pair.get_data())


def add_specialist_database(flyweights_factory: FlyweightFactory,
                            company: str, position: str,
                            name:str, passport: str):
    print()
    flyweight = flyweights_factory.get_flyweight(Shared(company, position))
    flyweight.process(Unique(name, passport))


if __name__ == '__main__':

    shared_list: List[Shared] = (
        Shared("Microsoft", "manager"),
        Shared("Google", "manager"),
        Shared("Google", "developer"),
        Shared("Apple", "developer")
    )

    flyweight_factory = FlyweightFactory(shared_list)
    flyweight_factory.list_flyweights()

    add_specialist_database(flyweight_factory,
                            "Google",
                            "developer",
                            "George",
                            "12345")

    add_specialist_database(flyweight_factory,
                            "Apple",
                            "manager",
                            "Mark",
                            "67890")

    flyweight_factory.list_flyweights()