mirror of
https://github.com/Hestia-Homes/Model.git
synced 2026-06-08 11:17:27 +00:00
27 lines
554 B
Python
27 lines
554 B
Python
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
|