def main():
# Get input parameters
parser = argparse.ArgumentParser()
parser.add_argument("-a",
"--action",
default='',
choices=[
'run', 'list', 'init', 'add', 'config', 'getconf',
'del', 'update', 'getEpisode', 'resetKeywords',
'resetAllKeywords', 'search', 'push', 'logs'
],
help='action triggered by the script')
parser.add_argument(
"-c",
"--config",
default=myConstants.CONFIG_FILE,
help='indicates the configuration file location. By default:' +
myConstants.CONFIG_FILE)
parser.add_argument(
"-s",
"--seriefile",
default=myConstants.SERIES_FILE,
help='indicates the series list file location. By default:' +
myConstants.SERIES_FILE)
parser.add_argument("-v",
"--verbosity",
action="store_true",
help='maximum output verbosity')
parser.add_argument("--admin", action="store_true", help='admin mode')
parser.add_argument("--arg",
default='',
help='arguments for bash execution')
args = parser.parse_args()
# Manage verbosity level
if args.verbosity:
logging.basicConfig(level=logging.DEBUG)
logging.info('TSW started in verbosity mode')
# Parse arguments
if args.arg != '':
try:
arg = json.loads(args.arg)
except:
print(
json.dumps({
'rtn': '415',
'error': messages.returnCode['415']
}))
sys.exit()
else:
arg = {}
# Initialize more data
migrateFiles.migrate()
m = TSWmachine(args.admin, args.verbosity)
if args.admin:
logging.info('TSW loaded in admin mode')
if args.action != 'init':
logging.debug('Loading of conffile: %s', args.config)
logging.debug('Loading of seriefile: %s', args.seriefile)
result = m.openFiles(args.config, args.seriefile)
if result['rtn'] != '200':
print(json.dumps(result))
sys.exit()
if args.action == 'init':
result = m.createConf(args.config, arg['conf'])
print(json.dumps(result))
sys.exit()
if args.action == 'run':
m.run()
sys.exit()
if args.action == 'list':
if 'ids' not in arg.keys():
arg['ids'] = 'all'
if 'load_tvdb' not in arg.keys():
arg['load_tvdb'] = False
print(
json.dumps(
m.getSeries(arg['ids'],
json_c=True,
load_tvdb=arg['load_tvdb'])))
sys.exit()
if args.action == 'add':
if 'emails' not in arg.keys():
arg['emails'] = []
print(json.dumps(m.addSeries(arg['id'], arg['emails'])))
sys.exit()
if args.action == 'search':
if 'pattern' not in arg.keys():
print(json.dumps({'rtn': 415, 'error': 'Blank search'}))
sys.exit()
print(json.dumps(m.search(arg['pattern'])))
sys.exit()
if args.action == 'update':
if 'param' not in arg.keys():
arg['param'] = {}
print(json.dumps(m.setSerie(arg['id'], arg['param'], json_c=True)))
sys.exit()
if args.action == 'resetKeywords':
print(json.dumps(m.resetKeywords(arg['id'])))
sys.exit()
if args.action == 'resetAllKeywords':
print(json.dumps(m.resetAllKeywords()))
sys.exit()
if args.action == 'config':
result = m.setConf(arg['conf'])
if (result['rtn'] != '200'):
print(json.dumps(result))
else:
print(json.dumps(m.getConf()))
sys.exit()
if args.action == 'getconf':
if 'arg' not in locals():
arg = {}
if 'conf' not in arg.keys():
arg['conf'] = 'all'
print(json.dumps(m.getConf(arg['conf'])))
sys.exit()
if args.action == 'del':
print(json.dumps(m.delSerie(arg['id'])))
sys.exit()
if args.action == 'getEpisode':
print(
json.dumps(m.getEpisode(arg['id'], arg['season'], arg['episode'])))
sys.exit()
if args.action == 'push':
print(json.dumps(m.pushTorrent(arg['id'], arg['filepath'])))
sys.exit()
if args.action == 'logs':
print(json.dumps(m.logs(json_c=True)))
sys.exit()
print(
json.dumps({
'rtn': '400',
'error': messages.returnCode['400'].format(args.action)
}))
sys.exit()
def main():
# Get input parameters
parser = argparse.ArgumentParser()
parser.add_argument(
"-a",
"--action",
default='run',
choices=['run', 'list', 'init', 'add','config','getconf','del','edit'],
help='action triggered by the script'
)
parser.add_argument(
"-c",
"--config",
default=myConstants.CONFIG_FILE,
help='indicates the configuration file location. By default:'+myConstants.CONFIG_FILE
)
parser.add_argument(
"-s",
"--seriefile",
default=myConstants.SERIES_FILE,
help='indicates the series list file location. By default:'+myConstants.SERIES_FILE
)
parser.add_argument(
"-v",
"--verbosity",
action="store_true",
help='maximum output verbosity'
)
parser.add_argument(
"--arg",
default='',
help='arguments for bash execution'
)
args = parser.parse_args()
# Manage verbosity level
if args.verbosity:
logging.basicConfig(level=logging.DEBUG)
verbose = True
else:
verbose = False
logging.info('SERIES started in verbosity mode')
global arg;
if args.arg != '':
Prompt.arg = args.arg.split(',')
migrateFiles.migrate()
# Initialize more data
m = TSWmachine(True,verbose)
if args.action != 'init':
logging.debug('Loading of conffile: %s', args.config)
logging.debug('Loading of seriefile: %s', args.seriefile)
result = m.openFiles(args.config, args.seriefile)
if result['rtn']!='200':
logging.debug('Fail loading config file: %s', result['error'])
print("Please first use tvShowWatch --action init")
sys.exit()
else:
result = m.createConf(args.config)
if result['rtn']!='200':
print('Error during creation of the configuration file: '+result['error'])
sys.exit()
action_fct = {
'list':action_list,
'run':action_run,
'add':action_add,
'init':action_reset,
'config':action_config,
'del':action_del,
'edit':action_edit,
'getconf':action_getconf
}
# Call for action
logging.debug('Action from the parameter: %s', args.action)
fct = action_fct[args.action]
logging.debug('Action function: %s', fct)
fct(m)
def __init__(self, debug=False):
self.debug = debug != False
self.auth = False
self.m = TSWmachine(True, False)
