manage.py

if __name__ == "__main__":
	import logSetup
	logSetup.initLogging()


# Shut up fucking annoying psycopg2 vomit every exec.
import warnings
from sqlalchemy import exc as sa_exc
warnings.filterwarnings("ignore", category=UserWarning, module='psycopg2')
warnings.simplefilter("ignore", category=sa_exc.SAWarning)

import inspect
import sys
import common.management.WebMirrorManage
import common.management.CrNManage
import common.management.RawMirrorManage
import common.management.Misc
import common.management.Testing

import common.Tools.WebFictionGuide

import WebMirror.OfflineFilters.offline_filters
import WebMirror.TimedTriggers.TriggerManage

# Site-specific management stuff
import WebMirror.management.GravityTalesManage
import WebMirror.management.SpcnetTvManage
import WebMirror.management.RssManage
import WebMirror.management.Testing

# Moar
import WebMirror.management.DbManage
import WebMirror.management.FeedDbManage
import WebMirror.management.UrlManage
import WebMirror.OfflineFilters.NewNetlocTracker

import RawArchiver.TimedTriggers.TriggerManage


func_prefix = "exposed_"

SCANNED_MODULES = [
	common.management.WebMirrorManage,
	common.management.RawMirrorManage,
	common.management.CrNManage,
	common.management.Misc,
	common.management.Testing,

	common.Tools.WebFictionGuide,

	WebMirror.OfflineFilters.offline_filters,
	WebMirror.TimedTriggers.TriggerManage,

	WebMirror.management.GravityTalesManage,
	WebMirror.management.SpcnetTvManage,

	WebMirror.management.FeedDbManage,
	WebMirror.management.DbManage,
	WebMirror.management.RssManage,
	WebMirror.management.UrlManage,
	WebMirror.management.Testing,

	WebMirror.OfflineFilters.NewNetlocTracker,

	RawArchiver.TimedTriggers.TriggerManage,

]

def load_functions():
	print("Loading functions")
	ret = {}
	for module in SCANNED_MODULES:
		for name, member in inspect.getmembers(module):
			if inspect.isfunction(member) and name.startswith(func_prefix):
				sname = name[len(func_prefix):]
				assert sname not in ret, "Duplicate management functions named: '%s'" % name
				ret[sname] = member
	print("Found %s functions" % len(ret))
	return ret


def print_func(name, func):

	doc = inspect.getdoc(func)
	sig = inspect.signature(func)

	if not doc or (doc and not doc.strip()):
		print("    {} -> {}".format(name.ljust(25), "UNDOCUMENTED"))
		if not sig.parameters:
			print("        No arguments")
		else:
			print("        Args: {}".format(sig))

	else:
		print("    {}".format(name))
		if not sig.parameters:
			print("        No arguments")
		else:
			print("        Args: {}".format(sig))
		doclines = doc.splitlines()
		for line in doclines:
			print("            -> {}".format(line))
	print()

def print_help():
	print("ReadableWebProxy Management CLI Interface!")
	print("Available functions:")
	farr = load_functions()
	names = list(farr.keys())
	names.sort()

	for name in names:
		print_func(name, farr[name])

def try_call(func, args):
	'''
	Try to call function `func` with passed array of arguments `args`.
	Validates that arguments args are of the correct length.
	'''


	sig = inspect.signature(func)

	if len(sig.parameters) == 0 and len(args) == 0:
		print("No params required: ", func)
		func()
		print("Called!")
		return True

	if len(sig.parameters) == len(args):
		print("Matching param count: ", func)
		func(*args)
		return True

	req_params = [parm for parm in sig.parameters if sig.parameters[parm].default == inspect.Parameter.empty]
	if len(args) >= len(req_params) and len(args) <= len(sig.parameters):
		print("Partial coverage of arguments, including all required: ", args)
		func(*args)
		return True

	return False

def call_func(args):
	print("Looking for function callable with params: '{}'".format(args))
	fname = args[0]
	farr = load_functions()
	if not fname in farr:
		return False

	print("Function signature: '%s'" % (inspect.getsourcelines(farr[fname])[0][0].strip(), ))
	print("Found function %s, from file %s, line %s" % (farr[fname], inspect.getsourcefile(farr[fname]), inspect.findsource(farr[fname])[1]))

	return try_call(farr[fname], args[1:])


def go():
	if len(sys.argv) > 1:
		have = call_func(sys.argv[1:])
		if not have:
			print_help()
		else:
			return
	else:
		print_help()

if __name__ == "__main__":
	go()