summaryrefslogtreecommitdiff
path: root/megapixels/cli_faiss.py
blob: 9953d9b72fc11c2c61aa16e55081fd5dcab1d407 (plain)
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
# --------------------------------------------------------
# add/edit commands in commands/faiss directory
# --------------------------------------------------------

import click

from app.settings import app_cfg as cfg
from app.utils import logger_utils
from app.models.click_factory import ClickSimple

# click cli factory
cc = ClickSimple.create(cfg.DIR_COMMANDS_FAISS)

# --------------------------------------------------------
# CLI
# --------------------------------------------------------
@click.group(cls=cc, chain=False)
@click.option('-v', '--verbose', 'verbosity', count=True, default=4, 
  show_default=True,
  help='Verbosity: -v DEBUG, -vv INFO, -vvv WARN, -vvvv ERROR, -vvvvv CRITICAL')
@click.pass_context
def cli(ctx, **kwargs):
  """\033[1m\033[94mMegaPixels: FAISS Data Scripts\033[0m                                                
  """
  ctx.opts = {}
  # init logger
  logger_utils.Logger.create(verbosity=kwargs['verbosity'])



# --------------------------------------------------------
# Entrypoint
# --------------------------------------------------------
if __name__ == '__main__':
    cli()