kopia lustrzana https://github.com/glidernet/ogn-python
35 wiersze
963 B
Python
35 wiersze
963 B
Python
from ogn.commands.dbutils import engine, session
|
|
from ogn.model import Base, AddressOrigin
|
|
from ogn.utils import get_ddb
|
|
from ogn.collect.database import update_devices
|
|
|
|
from manager import Manager
|
|
manager = Manager()
|
|
|
|
|
|
@manager.command
|
|
def init():
|
|
"""Initialize the database."""
|
|
|
|
Base.metadata.create_all(engine)
|
|
print("Done.")
|
|
|
|
|
|
@manager.command
|
|
def import_ddb():
|
|
"""Import registered devices from the DDB."""
|
|
|
|
print("Import registered devices fom the DDB...")
|
|
counter = update_devices(session, AddressOrigin.ogn_ddb, get_ddb())
|
|
print("Imported %i devices." % counter)
|
|
|
|
|
|
@manager.command
|
|
def import_file(path='tests/custom_ddb.txt'):
|
|
"""Import registered devices from a local file."""
|
|
# (flushes previously manually imported entries)
|
|
|
|
print("Import registered devices from '{}'...".format(path))
|
|
counter = update_devices(session, AddressOrigin.user_defined, get_ddb(path))
|
|
print("Imported %i devices." % counter)
|