# -*- coding: utf-8 -*-
"""
TODO: Rename to wbia/init/commands.py
TODO; remove params module
"""
from __future__ import absolute_import, division, print_function
import utool as ut
import sys
from wbia import constants as const
from wbia import params
from wbia.other import ibsfuncs
from wbia.init import sysres
from os.path import join
print, rrr, profile = ut.inject2(__name__)
[docs]def vdq(dbdir):
"""view directory and quit"""
_ibsdb = const.PATH_NAMES._ibsdb
ut.util_cplat.view_directory(join(dbdir, _ibsdb))
sys.exit(0)
[docs]def vdd(ibs):
' view data dir '
ut.util_cplat.view_directory(ibs.dbdir)
[docs]def vwd():
""" view work dir """
ut.util_cplat.view_directory(sysres.get_workdir())
# def preload_convert_hsdb(dbdir):
# """ Convert the database before loading (A bit hacky) """
# from wbia.dbio import ingest_hsdb
# ingest_hsdb.convert_hsdb_to_wbia(dbdir, force_delete=params.args.force_delete)
[docs]def preload_commands(dbdir, **kwargs):
""" Preload commands work with command line arguments and global caches """
# print('[main_cmd] preload_commands')
if params.args.dump_argv:
print(ut.repr2(vars(params.args), sorted_=False))
if params.args.dump_global_cache:
ut.global_cache_dump() # debug command, dumps to stdout
if params.args.set_workdir is not None:
sysres.set_workdir(params.args.set_workdir)
if params.args.get_workdir:
print(' Current work dir = %s' % sysres.get_workdir())
# if params.args.logdir is not None:
# sysres.set_logdir(params.args.logdir)
if params.args.get_logdir:
print(' Current local log dir = %s' % (sysres.get_logdir_local(),))
print(' Current global log dir = %s' % (sysres.get_logdir_global(),))
if params.args.view_logdir:
ut.view_directory(sysres.get_logdir_local())
ut.view_directory(sysres.get_logdir_global())
if params.args.view_logdir_local:
ut.view_directory(sysres.get_logdir_local())
if params.args.view_logdir_global:
ut.view_directory(sysres.get_logdir_local())
if ut.get_argflag('--vwd'):
vwd()
if ut.get_argflag('--vdq'):
print('got arg --vdq')
vdq(dbdir)
if kwargs.get('delete_ibsdir', False):
ibsfuncs.delete_wbia_database(dbdir)
if params.args.preload_exit:
print('[main_cmd] preload exit')
sys.exit(0)
[docs]def postload_commands(ibs, back):
"""
Postload commands deal with a specific wbia database
wbia --db PZ_MTEST --occur "*All Images" --query 1
wbia --db PZ_MTEST --occur "*All Images" --query-intra
"""
if ut.NOT_QUIET:
print('\n[main_cmd] postload_commands')
if params.args.view_database_directory:
print('got arg --vdd')
vdd(ibs)
if params.args.set_default_dbdir:
sysres.set_default_dbdir(ibs.get_dbdir())
if params.args.update_query_cfg is not None:
# Set query parameters from command line using the --cfg flag
cfgdict = ut.parse_cfgstr_list(params.args.update_query_cfg)
print('Custom cfgdict specified')
print(ut.repr2(cfgdict))
ibs.update_query_cfg(**cfgdict)
if params.args.edit_notes:
ut.editfile(ibs.get_dbnotes_fpath(ensure=True))
if params.args.delete_cache:
ibs.delete_cache()
if params.args.delete_cache_complete:
ibs.delete_cache(delete_imagesets=True)
if params.args.delete_query_cache:
ibs.delete_qres_cache()
if params.args.set_all_species is not None:
ibs._overwrite_all_annot_species_to(params.args.set_all_species)
if params.args.dump_schema:
ibs.db.print_schema()
if ut.get_argflag('--ipynb'):
back.launch_ipy_notebook()
select_imgsetid = ut.get_argval(
('--select-imgsetid', '--imgsetid', '--occur', '--gsid'), None
)
if select_imgsetid is not None:
print('\n+ --- CMD SELECT IMGSETID=%r ---' % (select_imgsetid,))
# Whoa: this doesnt work. weird.
# back.select_imgsetid(select_imgsetid)
# This might be the root of gui problems
# back.front._change_imageset(select_imgsetid)
back.front.select_imageset_tab(select_imgsetid)
print('L ___ CMD SELECT IMGSETID=%r ___\n' % (select_imgsetid,))
# Send commands to GUIBack
if params.args.select_aid is not None:
if back is not None:
try:
ibsfuncs.assert_valid_aids(ibs, (params.args.select_aid,))
except AssertionError:
print('Valid RIDs are: %r' % (ibs.get_valid_aids(),))
raise
back.select_aid(params.args.select_aid)
if params.args.select_gid is not None:
back.select_gid(params.args.select_gid)
if params.args.select_nid is not None:
back.select_nid(params.args.select_nid)
select_name = ut.get_argval('--select-name')
if select_name is not None:
import wbia.gui.guiheaders as gh
back.ibswgt.select_table_indicies_from_text(
gh.NAMES_TREE, select_name, allow_table_change=True
)
if ut.get_argflag(('--intra-occur-query', '--query-intra-occur', '--query-intra')):
back.special_query_funcs['intra_occurrence'](cfgdict={'use_k_padding': False})
qaid_list = ut.get_argval(('--query-aid', '--query'), type_=list, default=None)
if qaid_list is not None:
# qaid_list = params.args.query_aid
# fix stride case
if len(qaid_list) == 1 and isinstance(qaid_list[0], tuple):
qaid_list = list(qaid_list[0])
daids_mode = ut.get_argval(
'--daids-mode', type_=str, default=const.VS_EXEMPLARS_KEY
)
back.compute_queries(qaid_list=qaid_list, daids_mode=daids_mode, ranks_top=10)
if ut.get_argflag('--inc-query'):
back.incremental_query()
if ut.get_argflag(('--dbinfo', '--display_dbinfo')):
back.display_dbinfo()
pass
aidcmd = ut.get_argval('--aidcmd', default=None)
aid = ut.get_argval('--aid', type_=int, default=1)
if aidcmd:
# aidcmd = 'Interact image'
metadata = ibs.get_annot_lazy_dict(aid)
annot_context_options = metadata['annot_context_options']
aidcmd_dict = dict(annot_context_options)
print('aidcmd_dict = %s' % (ut.repr3(aidcmd_dict),))
command = aidcmd_dict[aidcmd]
command()
# import utool
# utool.embed()
# back.start_web_server_parallel()
if ut.get_argflag('--start-web'):
back.start_web_server_parallel()
if ut.get_argflag('--name-tab'):
from wbia.gui.guiheaders import NAMES_TREE
back.front.set_table_tab(NAMES_TREE)
view = back.front.views[NAMES_TREE]
model = view.model()
view._set_sort(model.col_name_list.index('nAids'), col_sort_reverse=True)
if ut.get_argflag('--graph'):
back.make_qt_graph_interface()
screengrab_fpath = ut.get_argval('--screengrab')
if screengrab_fpath:
from wbia.guitool.__PYQT__.QtGui import QPixmap
from PyQt4.QtTest import QTest
from PyQt4.QtCore import Qt
fpath = ut.truepath(screengrab_fpath)
import wbia.guitool
# ut.embed()
timer2 = guitool.__PYQT__.QtCore.QTimer()
done = [1000]
def delayed_screenshot_func():
if done[0] == 500:
# back.mainwin.menubar.triggered.emit(back.mainwin.menuFile)
print('Mouseclick')
QTest.mouseClick(back.mainwin.menuFile, Qt.LeftButton)
# This works
# QTest.mouseClick(back.front.import_button, Qt.LeftButton)
if done[0] == 1:
timer2.stop()
print('screengrab to %r' % (fpath,))
screenimg = QPixmap.grabWindow(back.mainwin.winId())
screenimg.save(fpath, 'jpg')
ut.startfile(fpath)
print('lub dub2')
done[0] -= 1
return None
CLICK_FILE_MENU = True
if CLICK_FILE_MENU:
# ut.embed()
# QTest::keyClick(menu, Qt::Key_Down)
pass
timer2.delayed_screenshot_func = delayed_screenshot_func
timer2.timeout.connect(timer2.delayed_screenshot_func)
timer2.start(1)
back.mainwin.timer2 = timer2
guitool.activate_qwindow(back.mainwin)
# QPixmap.grabWindow(back.mainwin.winId()).save(fpath, 'jpg')
# ut.startfile(fpath)
# ut.embed()
pass
if params.args.postload_exit:
print('[main_cmd] postload exit')
sys.exit(0)