import pandas as pd
from model_data.utils import setup_logger

logger = setup_logger()


class OpenUprnClient:
    COLUMN_NAMES = [

    ]

    def __init__(self, path, uprns):
        self.path = path
        self.uprns = [int(x) for x in uprns]
        self.data = None

    def read(self):
        """
        This methodology is placeholder, while data sits localls
        :return:
        """
        logger.info("Reading in open uprn data")

        df = pd.read_csv(self.path)
        df = df[df["UPRN"].isin(self.uprns)]

        self.data = df