38 lines
1.4 KiB
Python
38 lines
1.4 KiB
Python
|
|
from . import crud, main, schemas, auth_helper
|
|
from decouple import config
|
|
from .database import SessionLocal
|
|
from datetime import timedelta
|
|
|
|
db = SessionLocal()
|
|
|
|
def init_user():
|
|
user = schemas.UserCreate(email="hisham@banafa.com.sa",
|
|
username="Hesham",
|
|
password=config('first_user_pass'))
|
|
user_exists = crud.get_user_by_email(db, user.email)
|
|
if user_exists: return
|
|
crud.create_user(db, user)
|
|
token = auth_helper.create_access_token(data={"sub": user.username}, expires_delta=timedelta(minutes=15))
|
|
res = crud.set_user_last_token(db, user.username, token)
|
|
if not res: print("Failed to add initial token")
|
|
|
|
def init_door():
|
|
iot_door = schemas.IotEntityCreate(bluetooth_mac="ff:ff:ff:ff",
|
|
description="Iot Lab Door")
|
|
door_exists = crud.get_iot_entity_by_bluetooth_mac(db, iot_door.bluetooth_mac)
|
|
if door_exists: return
|
|
crud.create_iot_entity(db, iot_door)
|
|
|
|
def init_monitor():
|
|
iot_monitor = schemas.IotEntityCreate(bluetooth_mac="ff:ff:00:ff",
|
|
description="Iot Lab Monitor")
|
|
monitor_exists = crud.get_iot_entity_by_bluetooth_mac(db, iot_monitor.bluetooth_mac)
|
|
if monitor_exists: return
|
|
crud.create_iot_entity(db, iot_monitor)
|
|
|
|
def init():
|
|
init_user()
|
|
init_door()
|
|
init_monitor()
|
|
|