42 lines
1.1 KiB
Python
42 lines
1.1 KiB
Python
|
from uuid import uuid4
|
||
|
|
|
||
|
|
import click
|
||
|
|
from boltons.urlutils import URL
|
||
|
|
|
||
|
|
from ereuse_devicehub.db import db
|
||
|
|
from ereuse_devicehub.resources.agent.models import Person
|
||
|
|
from ereuse_devicehub.resources.inventory.model import Inventory
|
||
|
|
from ereuse_devicehub.resources.user.models import User
|
||
|
|
|
||
|
|
|
||
|
|
class InitDatas:
|
||
|
|
def __init__(self, app) -> None:
|
||
|
|
super().__init__()
|
||
|
|
self.app = app
|
||
|
|
self.schema = app.config.get('DB_SCHEMA')
|
||
|
|
self.app.cli.command(
|
||
|
|
'initdata', short_help='Save a minimum structure of datas.'
|
||
|
|
)(self.run)
|
||
|
|
|
||
|
|
def run(self):
|
||
|
|
inv = Inventory(
|
||
|
|
id=self.schema,
|
||
|
|
name="usody",
|
||
|
|
tag_provider=URL('http://localhost:8081'),
|
||
|
|
tag_token=uuid4(),
|
||
|
|
org_id=uuid4(),
|
||
|
|
)
|
||
|
|
|
||
|
|
db.session.add(inv)
|
||
|
|
db.session.commit()
|
||
|
|
|
||
|
|
email = 'user@dhub.com'
|
||
|
|
password = '1234'
|
||
|
|
name = 'user'
|
||
|
|
|
||
|
|
user = User(email=email, password=password)
|
||
|
|
user.individuals.add(Person(name=name))
|
||
|
|
db.session.add(user)
|
||
|
|
|
||
|
|
db.session.commit()
|