clened recommendation files

This commit is contained in:
MauricioGarciaS 2022-11-29 11:17:59 +01:00
parent dcc8ef5fb0
commit 77536c3153
11 changed files with 12 additions and 42 deletions

View file

@ -1,42 +0,0 @@
import logging
from apscheduler.schedulers.asyncio import AsyncIOScheduler
from fastapi import FastAPI
# from fastapi_utils.tasks import repeat_every
from utils import events_queue
from utils import pg_client
from utils import schemas_ee
app = FastAPI()
app.schedule = AsyncIOScheduler()
@app.get('/')
def home():
return '<h1>This is a title</h1>'
@app.get('/value/{value}')
@app.put('/value/{value}')
def number(value: int):
logging.info(f'> {value} as input. Testing queue with pg')
d = {'timestamp': 23786, 'action': 'action', 'source': 'source', 'category': 'cat', 'data': {}}
events = schemas_ee.SignalsSchema
event = events.parse_obj(d)
events_queue.global_queue.put((value, 0, event))
@app.on_event("startup")
async def startup():
await pg_client.init()
await events_queue.init(test=False)
app.schedule.start()
@app.schedule.scheduled_job("interval", seconds=60*1)
def clean_up():
events_queue.global_queue.force_flush()
@app.on_event("shutdown")
async def shutdown():
await events_queue.terminate()
await pg_client.terminate()

View file

@ -0,0 +1 @@
docker-compose down --volumes --rmi all

11
ee/recommendation/run.sh Normal file
View file

@ -0,0 +1,11 @@
echo 'Setting up required modules..'
mkdir scripts
mkdir plugins
mkdir logs
mkdir scripts/utils
cp ../../api/chalicelib/utils/pg_client.py scripts/utils
cp ../api/chalicelib/utils/ch_client.py scripts/utils
echo 'Building containers...'
docker-compose up airflow-init
echo 'Running containers...'
docker-compose up