1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
|
"""
Click processor factory
- Inspired by and used code from @wiretapped's HTSLAM codebase
- In particular the very useful
"""
import os
import sys
from os.path import join
from pathlib import Path
import os
from os.path import join
import sys
from functools import update_wrapper, wraps
import itertools
from pathlib import Path
from glob import glob
import importlib
import logging
import click
from app.settings import app_cfg as cfg
# --------------------------------------------------------
# Click Group Class
# --------------------------------------------------------
# set global variable during parent class create
dir_plugins = None # set in create
class ClickComplex:
"""Wrapper generator for custom Click CLI's based on LR's coroutine"""
def __init__(self):
pass
class CustomGroup(click.Group):
#global dir_plugins # from CliGenerator init
# lists commands in plugin directory
def list_commands(self, ctx):
global dir_plugins # from CliGenerator init
rv = list(self.commands.keys())
fp_cmds = [Path(x) for x in Path(dir_plugins).iterdir() \
if str(x).endswith('.py') \
and '__init__' not in str(x)]
for fp_cmd in fp_cmds:
try:
assert fp_cmd.name not in rv, "[-] Error: {} can't exist in cli.py and {}".format(fp_cmd.name)
except Exception as ex:
logging.getLogger('app').error('{}'.format(ex))
rv.append(fp_cmd.stem)
rv.sort()
return rv
# Complex version: gets commands in directory and in this file
# Based on code from @wiretapped + HTSLAM
def get_command(self, ctx, cmd_name):
global dir_plugins
if cmd_name in self.commands:
return self.commands[cmd_name]
ns = {}
fpp_cmd = Path(dir_plugins, cmd_name + '.py')
fp_cmd = fpp_cmd.as_posix()
if not fpp_cmd.exists():
sys.exit('[-] {} file does not exist'.format(fpp_cmd))
code = compile(fpp_cmd.read_bytes(), fp_cmd, 'exec')
try:
eval(code, ns, ns)
except Exception as ex:
logging.getLogger('vframe').error('exception: {}'.format(ex))
@click.command()
def _fail():
raise Exception('while loading {}'.format(fpp_cmd.name))
_fail.short_help = repr(ex)
_fail.help = repr(ex)
return _fail
if 'cli' not in ns:
sys.exit('[-] Error: {} does not contain a cli function'.format(fp_cmd))
return ns['cli']
@classmethod
def create(self, dir_plugins_local):
global dir_plugins
dir_plugins = dir_plugins_local
return self.CustomGroup
class ClickSimple:
"""Wrapper generator for custom Click CLI's"""
def __init__(self):
pass
class CustomGroup(click.Group):
#global dir_plugins # from CliGenerator init
# lists commands in plugin directory
def list_commands(self, ctx):
global dir_plugins # from CliGenerator init
rv = list(self.commands.keys())
fp_cmds = [Path(x) for x in Path(dir_plugins).iterdir() \
if str(x).endswith('.py') \
and '__init__' not in str(x)]
for fp_cmd in fp_cmds:
assert fp_cmd.name not in rv, "[-] Error: {} can't exist in cli.py and {}".format(fp_cmd.name)
rv.append(fp_cmd.stem)
rv.sort()
return rv
# Complex version: gets commands in directory and in this file
# from HTSLAM
def get_command(self, ctx, cmd_name):
global dir_plugins # from CliGenerator init
if cmd_name in self.commands:
return self.commands[cmd_name]
ns = {}
fpp_cmd = Path(dir_plugins, cmd_name + '.py')
fp_cmd = fpp_cmd.as_posix()
if not fpp_cmd.exists():
sys.exit('[-] {} file does not exist'.format(fpp_cmd))
code = compile(fpp_cmd.read_bytes(), fp_cmd, 'exec')
try:
eval(code, ns, ns)
except Exception as ex:
logging.getLogger('vframe').error('exception: {}'.format(ex))
@click.command()
def _fail():
raise Exception('while loading {}'.format(fpp_cmd.name))
_fail.short_help = repr(ex)
_fail.help = repr(ex)
return _fail
if 'cli' not in ns:
sys.exit('[-] Error: {} does not contain a cli function'.format(fp_cmd))
return ns['cli']
@classmethod
def create(self, dir_plugins_local):
global dir_plugins
dir_plugins = dir_plugins_local
return self.CustomGroup
|
