def create_new_snapshot(params: dict):
with flask_app.app_context():
snap = Snapshot(**params)
db.session.add(snap)
db.session.commit()
db.session.close()
return snap
def create_new_option(amount: float, strike_price: int, period: int,
option_type: int, market: str) -> Dict:
with flask_app.app_context():
status_code_id = OPTIONS_ESTIMATE
option = Option(
amount=amount,
execution_strategy_id=AUTO_ITM_EXECUTION,
strike_price=strike_price,
period=period,
market=market,
date_modified=datetime.utcnow(),
date_created=datetime.utcnow(),
option_type=option_type,
expiration_date=datetime.utcnow() + timedelta(seconds=period),
status_code_id=status_code_id,
)
db.session.add(option)
db.session.commit()
_id = option.id
db.session.close()
return {
"option_id": _id,
"amount": amount,
"strike_price": strike_price,
"period": period,
"option_type": option_type,
"market": market,
"status_code": status_code_id,
}
def update_agent(agent_address: str, params: dict) -> Option:
with flask_app.app_context():
agent = db.session.query(Agent).filter(
Agent.address == agent_address).one()
for key, value in params.items():
setattr(agent, key, value)
db.session.merge(agent)
db.session.commit()
db.session.close()
return agent
def update_option(option_db_id: int, params: dict) -> Option:
with flask_app.app_context():
option = db.session.query(Option).filter(
Option.id == option_db_id).one()
for key, value in params.items():
setattr(option, key, value)
db.session.merge(option)
db.session.commit()
db.session.close()
return option
def create_agent(agent_address: str, params: dict) -> Option:
with flask_app.app_context():
agents = (db.session.query(Agent).filter(
Agent.address == agent_address).all())
if len(agents) == 0:
agent = Agent(
address=agent_address,
date_created=datetime.now(),
date_updated=datetime.now(),
status="running",
)
db.session.merge(agent)
db.session.commit()
db.session.close()
return agent.id
if len(agents) == 1:
return agents[0].id
raise ValueError("Inexpected number of agents in db!")
def add_data():
"""Create the initial state for the agent"""
with flask_app.app_context():
db.create_all()
db.session.merge(
ExecutionStrategy(id=LIMIT_ORDER, description="limit_order"))
db.session.merge(
ExecutionStrategy(id=AUTO_ITM_EXECUTION,
description="auto_itm_execution"))
db.session.merge(
StatusCode(id=OPTIONS_ESTIMATE,
description="options_estimate"))
db.session.merge(
StatusCode(id=PENDING_PLACEMENT,
description="pending_placement"))
db.session.merge(StatusCode(id=PLACING, description="placing"))
db.session.merge(StatusCode(id=OPEN, description="open"))
db.session.merge(StatusCode(id=CLOSED, description="closed"))
db.session.merge(StatusCode(id=FAILED, description="failed"))
db.session.merge(StatusCode(id=EXPIRED, description="expired"))
db.session.query(Option).delete()
db.session.commit()
db.session.close()
def get_option(option_id: int) -> Option:
with flask_app.app_context():
option = db.session.query(Option).filter(
Option.id == option_id).one()
db.session.close()
return option
def get_options():
with flask_app.app_context():
db.create_all()
results = db.session.query(Option).all()
db.session.close()
return results
def delete_options() -> bool:
with flask_app.app_context():
_ = db.session.query(Option).delete()
db.session.commit()
db.session.close()
return True