input args

Main.py

 from RotTable import *
 from Traj3D import *
 
+import argparse
+parser = argparse.ArgumentParser()
+parser.add_argument("--filename", help="input filename of DNA sequence")
+parser.parse_args()
+args = parser.parse_args()
 
 def main():
 
     rot_table = RotTable()
     traj = Traj3D()
 
+    if args.filename:
+	    # Read file
+	    lineList = [line.rstrip('\n') for line in open("plasmid_8k.fasta")]
+		# Formatting
+	    seq = ''.join(lineList[1:])
+	    traj.compute(seq, rot_table)
+    else:
         traj.compute("AAAGGATCTTCTTGAGATCCTTTTTTTCTGCGCGTAATCTGCTGCCAGTAAACGAAAAAACCGCCTGGGGAGGCGGTTTAGTCGAAGGTTAAGTCAG", rot_table)
 
     print(traj.getTraj())