__init__.py

from PythonVoiceCodingPlugin.queries.abstract import *
from PythonVoiceCodingPlugin.queries.argument import SelectArgument
from PythonVoiceCodingPlugin.queries.alternatives import SelectAlternative
from PythonVoiceCodingPlugin.queries.big_roi import SelectBigRoi
from PythonVoiceCodingPlugin.queries.paste_back import PasteBack
from PythonVoiceCodingPlugin.queries.collect_variable import CollectVariable
from PythonVoiceCodingPlugin.queries.collect_parameter import CollectParameter
from PythonVoiceCodingPlugin.queries.collect_module import CollectModule
from PythonVoiceCodingPlugin.queries.collect_function_name import CollectFunctionName
from PythonVoiceCodingPlugin.queries.collect_imported_value import CollectImportedValue
from PythonVoiceCodingPlugin.queries.collect_decorator import CollectDecorator
from PythonVoiceCodingPlugin.queries.collect_class_name import CollectClassName
from PythonVoiceCodingPlugin.queries.insert_item import InsertItem
from PythonVoiceCodingPlugin.queries.delete_alternatives import DeleteAlternatives
from PythonVoiceCodingPlugin.queries.swap_back import SwapBack
from PythonVoiceCodingPlugin.queries.select_part import SelectPart
from PythonVoiceCodingPlugin.queries.select_back import SelectBack
from PythonVoiceCodingPlugin.queries.remember_here import RememberHere




def get_query(query_description):
	index = query_description["command"]
	if index in ["collect_indexable"]:
		index = {
			"variable":"collect_variable",
			"parameter":"collect_parameter",
			"module":"collect_module",
			"import value":"collect_imported_value",
			"function name":"collect_function_name",
			"class name":"collect_class_name",
			"decorator":"collect_decorator",
		}[query_description["collectable"]]
	h = {
		"argument": SelectArgument,
		"alternative": SelectAlternative,
		"big_roi": SelectBigRoi,
		"paste_back": PasteBack,

		"collect_variable": CollectVariable,
		"collect_parameter":CollectParameter,
		"collect_module":CollectModule,
		"collect_imported_value": CollectImportedValue,
		"collect_function_name": CollectFunctionName,
		"collect_decorator":CollectDecorator,
		"collect_class_name":CollectClassName,


		"insert_item": InsertItem,
		"delete_alternatives":DeleteAlternatives,
		"swap_back": SwapBack,
		"select_part": SelectPart,
		"select_back": SelectBack,
		"remember_here": RememberHere,
	}
	return h[index]
	

def get_secondary_query(query_description):
	if "operation" not in query_description:
		return {}
	else:
		h={
			"paste":{
				"command":"paste_back",
				"format":1,
			},
			"delete":{
				"command":"delete_alternatives",
				"format":1,
				"color":0,
			},
			"swap":{
				"command":"swap_back",
				"format":1,
			},
			"edit":{
				"command":"alternative",
				"format":3,
				"color":0,
			},
		}
		return h[query_description["operation"]]