import requests

data = {
'collection_ids': [10],
 'bucket': 'fansifter-model-data',
 'file_key': 'cc56867d369622c60905d6767421bbbba3ec6384a91ff224fa6e0910c/c0d1ce5c-4da2-4a4e-ae22-8e2b2f0c8741-20210226-151300',
 'schema': 'cc56867d369622c60905d6767421bbbba3ec6384a91ff224fa6e0910c'
}

#endpoint = '/train_rfm_models'
#endpoint = '/train_rfm_bins'
endpoint = '/train_models'


url = f"http://localhost:12345{endpoint}"  # do modelling
r = requests.post(url, json=data)
print(repr(r.content))


def test_big_ml():
    data = {'collection_ids': [185], 'bucket': 'fansifter-model-data',
         'file_key': 'ca7a460d855e229dd0384001e1c9f5d6fb59381f967ba1b7ec36a7591/3c1e423d-816d-4777-9ee6-5e0990459646-20210130-141210',
         'schema': 'ca7a460d855e229dd0384001e1c9f5d6fb59381f967ba1b7ec36a7591'}
    url = 'http://big_ml_engine.fansifter.cloud/train_models'
    r = requests.post(url, json=data)
    print(repr(r.content))