This repository has been archived on 2024-05-31. You can view files and clone it, but cannot push or open issues or pull requests.
devicehub-teal/ereuse_devicehub/cli.py

89 lines
2.6 KiB
Python
Raw Normal View History

2019-01-23 15:55:04 +00:00
import os
import click.testing
import flask.cli
2023-03-21 16:31:43 +00:00
import ereuse_devicehub.ereuse_utils
2019-01-23 15:55:04 +00:00
from ereuse_devicehub.config import DevicehubConfig
from ereuse_devicehub.devicehub import Devicehub
2021-06-08 14:14:30 +00:00
import sys
2023-03-21 16:31:43 +00:00
2021-06-08 14:14:30 +00:00
sys.ps1 = '\001\033[92m\002>>> \001\033[0m\002'
2023-03-21 16:31:43 +00:00
sys.ps2 = '\001\033[94m\002... \001\033[0m\002'
2021-06-08 14:14:30 +00:00
2022-03-02 12:39:15 +00:00
import os, readline, atexit
2023-03-21 16:31:43 +00:00
2021-06-08 14:14:30 +00:00
history_file = os.path.join(os.environ['HOME'], '.python_history')
try:
2023-03-21 16:31:43 +00:00
readline.read_history_file(history_file)
2021-06-08 14:14:30 +00:00
except IOError:
2023-03-21 16:31:43 +00:00
pass
2021-06-08 14:14:30 +00:00
readline.parse_and_bind("tab: complete")
readline.parse_and_bind('"\e[5~": history-search-backward')
readline.parse_and_bind('"\e[6~": history-search-forward')
readline.parse_and_bind('"\e[5C": forward-word')
readline.parse_and_bind('"\e[5D": backward-word')
readline.parse_and_bind('"\e\e[C": forward-word')
readline.parse_and_bind('"\e\e[D": backward-word')
readline.parse_and_bind('"\e[1;5C": forward-word')
readline.parse_and_bind('"\e[1;5D": backward-word')
readline.set_history_length(100000)
atexit.register(readline.write_history_file, history_file)
2019-01-23 15:55:04 +00:00
2023-03-21 16:31:43 +00:00
2019-01-23 15:55:04 +00:00
class DevicehubGroup(flask.cli.FlaskGroup):
2019-02-04 12:01:36 +00:00
# todo users cannot make cli to use a custom db this way!
2019-01-23 15:55:04 +00:00
CONFIG = DevicehubConfig
def main(self, *args, **kwargs):
# todo this should be taken as an argument for the cli
inventory = os.environ.get('dhi')
if not inventory:
raise ValueError('Please do "export dhi={inventory}"')
self.create_app = self.create_app_factory(inventory)
return super().main(*args, **kwargs)
@classmethod
def create_app_factory(cls, inventory):
return lambda: Devicehub(inventory, config=cls.CONFIG())
def get_version(ctx, param, value):
if not value or ctx.resilient_parsing:
return
2023-03-21 16:31:43 +00:00
click.echo(
'Devicehub {}'.format(
ereuse_devicehub.ereuse_utils.version('ereuse-devicehub')
),
color=ctx.color,
)
flask.cli.get_version(ctx, param, value)
2023-03-21 16:31:43 +00:00
@click.option(
'--version',
help='Devicehub version.',
expose_value=False,
callback=get_version,
is_flag=True,
is_eager=True,
)
@click.group(
cls=DevicehubGroup,
context_settings=Devicehub.cli_context_settings,
add_version_option=False,
help="""Manages the Devicehub of the inventory {}.
Use 'export dhi=xx' to set the inventory that this CLI
manages. For example 'export dhi=db1' and then executing
'dh tag add' adds a tag in the db1 database. Operations
that affect the common database (like creating an user)
are not affected by this.
2023-03-21 16:31:43 +00:00
""".format(
os.environ.get('dhi')
),
)
2019-01-23 15:55:04 +00:00
def cli():
pass