problem6_repository.py

"""
Problem 6: Repository Pattern
"""

from abc import ABC, abstractmethod
import json


class User:
    def __init__(self, user_id, name, email):
        self.id = user_id
        self.name = name
        self.email = email


class UserRepository(ABC):
    @abstractmethod
    def get_by_id(self, user_id): pass

    @abstractmethod
    def get_all(self): pass

    @abstractmethod
    def add(self, user): pass

    @abstractmethod
    def update(self, user): pass

    @abstractmethod
    def delete(self, user_id): pass


class InMemoryUserRepository(UserRepository):
    def __init__(self):
        self.users = {}

    def get_by_id(self, user_id):
        return self.users.get(user_id)

    def get_all(self):
        return list(self.users.values())

    def add(self, user):
        self.users[user.id] = user

    def update(self, user):
        self.users[user.id] = user

    def delete(self, user_id):
        del self.users[user_id]


class UserService:
    def __init__(self, repository: UserRepository):
        self.repository = repository

    def create_user(self, user):
        self.repository.add(user)

    def list_users(self):
        return self.repository.get_all()


if __name__ == "__main__":
    repo = InMemoryUserRepository()
    service = UserService(repo)

    service.create_user(User(1, "Fara", "fara@email.com"))
    service.create_user(User(2, "Ali", "ali@email.com"))

    for user in service.list_users():
        print(user.name, user.email)