#!/usr/bin/env python

import sys
import signal
from tios import transmitter
from tios._version import __version__
import argparse

parser = argparse.ArgumentParser(description='Submit an MD simulation to *tios*', usage='tios-submit [-h] [--title TITLE] [--selection SELECTION] [--trate TRATE] job_string')
parser.add_argument('-t', '--title', 
                    help='descriptive title for the simulation')
parser.add_argument('-s', '--selection', default='all',
                    help='subset of atoms to be displayed on the web page')
parser.add_argument('-r', '--trate', type=float, default=1.0,
                    help='interval (in ps) between frames')
parser.add_argument('-V', '--version', action='version', version=__version__)
indx = 0
for i in range(1, len(sys.argv)):
    if i%2 == 1 and sys.argv[i][0] != '-':
        indx = i
        break

if len(sys.argv) == 2:
    args = parser.parse_args()
else:
    args = parser.parse_args(sys.argv[1:indx])

command = ' '.join(sys.argv[indx:])
tj = transmitter.from_command(command, title=args.title,
                           trate=args.trate, selection=args.selection,
                           protocol='Mongo')
print('Tios job {} registered.'.format(tj.id))
