49 lines
2 KiB
Python
49 lines
2 KiB
Python
from analysis.market_snapshot import Market, available_date_ranges
|
|
from db_layer.capturas_interface import capturas_interface
|
|
from db_layer.indices_interface import indices_interface
|
|
|
|
|
|
class IndexMM:
|
|
|
|
def __init__(self):
|
|
self.name = 'indexmm'
|
|
self.market = None
|
|
self.date = None
|
|
self.data = None
|
|
self.value = None
|
|
|
|
def calculate(self, market):
|
|
self.market = market
|
|
self.date = self.market.end_date
|
|
self.data = self.market.get_market_data()
|
|
|
|
data_coche_pequeno = {'count': self.data[self.data['tamano_categorico'] == 'coche pequeño'].shape[0],
|
|
'mean': self.data[self.data['tamano_categorico'] == 'coche pequeño']['precio'].mean()}
|
|
data_coche_grande = {'count': self.data[self.data['tamano_categorico'] == 'coche grande'].shape[0],
|
|
'mean': self.data[self.data['tamano_categorico'] == 'coche grande']['precio'].mean()}
|
|
data_coche_moto = {'count': self.data[self.data['tamano_categorico'] == 'coche y moto'].shape[0],
|
|
'mean': self.data[self.data['tamano_categorico'] == 'coche y moto']['precio'].mean()}
|
|
|
|
self.value = (((data_coche_grande['count'] * data_coche_grande['mean']) + (data_coche_moto['count'] * data_coche_moto['mean'])
|
|
+ (data_coche_pequeno['count'] * data_coche_pequeno['mean']))
|
|
/ (data_coche_grande['count'] + data_coche_moto['count'] + data_coche_pequeno['count']))
|
|
|
|
def get_data(self):
|
|
return {'name': self.name,
|
|
'date': self.date,
|
|
'value': self.value.item()}
|
|
|
|
|
|
if __name__ == '__main__':
|
|
for date_range in available_date_ranges:
|
|
market = Market()
|
|
market.load_market(capturas_interface.get_market_snapshot(date_range['start'], date_range['end']),
|
|
date_range=date_range)
|
|
market.clean_market('index')
|
|
|
|
index = IndexMM()
|
|
index.calculate(market)
|
|
|
|
indices_interface.write_index(index.get_data())
|
|
|
|
|