Entites and Repositories

smorgasbord.py

import pytest
import json


# ---------------------------- #
# lib/entities/entity.py
# ---------------------------- #

class Entity(object):

    FIELDS = []

    def __init__(self, data = {}):
        self.attributes = {}

        for key in self.FIELDS:
            self.attributes[key] = None

        for key in data:
            if key in self.FIELDS:
                self.attributes[key] = data[key]

    def __getattr__(self, name):
        if name in self.attributes:
            return self.attributes[name]
        else:
            raise AttributeError, name

# ---------------------------- #
# lib/entities/job_post_entity.py
# ---------------------------- #

from .entity import Entity

class JobPostEntity(Entity):

    FIELDS = [
        'description',
        'job_type',
    ]

# ---------------------------- #
# lib/repository/repository.py
# ---------------------------- #

class Repository:

    def __init__(self, database):
        self.database = database

    def add(self, entity):
        raise NotImplementedError

    def remove(self, entity):
        raise NotImplementedError

    def get(self, id):
        raise NotImplementedError

    def find(self):
        raise NotImplementedError

    def exists(self, id):
        raise NotImplementedError

# ---------------------------- #
# lib/repository/job_post_repository.py
# ---------------------------- #

from .repository import Repository
from ..entity.job_post_entity import JobPostEntity

class JobPostRepository(Repository):

   def get(self, id):
        sql = "SELECT * FROM job_posts WHERE id = %s"

        self.database.execute(sql, (id, ))
        data = self.database.fetch_as_dict()

        if data:
            attributes = {}
            attributes['description'] = data['description']
            attributes['job_type'] = data['job_type']

            return JobPostEntity(attributes)

    def add(self, job_post):
        if self.exists(job_post.id):
            self.update_job_post(job_post)
        else:
            self.insert_job_post(job_post)

            
class TestEntity():

    def test_first_name(self):
        entity = Entity(data = {"first_name": "James"})
        assert entity.first_name == "James"

    def test_last_name(self):
        entity = Entity(data = {"last_name": "Bond"})
        assert entity.last_name == "Bond"