def main():
args = parse_args()
print(args.workdir)
engine = create_engine(engine_file)
Job.metadata.create_all(engine)
Session = sessionmaker(bind=engine)
session = Session()
if args.workdir == [] and os.path.exists("workdir"): args.workdir = ["workdir"]
# Restore an existing workdir
if len(args.workdir):
for workdir in args.workdir:
job = get_job_from_workdir(session, workdir)
print job
if job.status == "Q" or job.status == "R" or job.status == "PD":
print("Job already running or queued")
return
cluster, shell = Clusters().get_cluster(job.cluster_name)
job = cluster.submit(shell, job, duration=args.duration, nodes=args.nodes, partition=args.partition, ntasks_per_node=args.ntasks_per_node)
status = cluster.get_status(shell, job)
job.status = status
if args.partition and job.partition != args.partition: job.partition = args.partition
if args.nodes and job.nodes != args.nodes: job.nodes = args.nodes
session.add(job)
session.commit()
# Create a brand-new workdir
else:
cluster, shell = Clusters().get_cluster(args.cluster)
job = create(args.topol, cluster, shell, args.jobname, args.duration, args.nodes, args.processes, args.script, args.partition, ntasks_per_node=args.ntasks_per_node)
assert(job)
print job
status = cluster.get_status(shell, job)
print status
if not job.cluster_id:
job = cluster.submit(shell, job)
print status
session.add(job)
session.commit()
