adding loader with historical data loading

This commit is contained in:
2025-12-05 11:54:58 +01:00
parent 9041af2979
commit a351af1dbe

View File

@@ -118,9 +118,27 @@ if __name__ == '__main__':
def __init__(self, backend_url: str): def __init__(self, backend_url: str):
SupabaseProvider.__init__(self) SupabaseProvider.__init__(self)
BackendAPIProvider.__init__(self, backend_url=backend_url) BackendAPIProvider.__init__(self, backend_url=backend_url)
class HistoricalProvider(SupabaseProvider, BackendAPIProvider):
def fetch_kafka_topic(self, topic: str) -> pd.DataFrame:
path = "/home/velocitatem/Documents/Projects/PHANTOM/experiments/collected_data/858c61ab-0a7f-4595-ae49-33f4365517b9/"
interactions_file = "messages(2).json"
prices_file = "messages(3).json"
if topic == "interactions":
data = pd.read_json(path + interactions_file)
elif topic == "price_logs":
pd.read_json(path + prices_file)
data = pd.read_json(path + (interactions_file if topic == "user-interactions" else prices_file))
data = [r['payload'] for r in data['value'].to_list()]
data = pd.DataFrame(data)
return data
# example run # example run
context = PipelineContext( context = PipelineContext(
provider=Provider(backend_url="http://localhost:5000"), provider=HistoricalProvider(),
store_mode='hotel', store_mode='hotel',
# 15 min not month # 15 min not month
window_size='15min', window_size='15min',