Skip to content
Snippets Groups Projects

refactor: update CLI

Merged Mate Balajti requested to merge fix_setup into main
5 files
+ 22
16
Compare changes
  • Side-by-side
  • Inline
Files
5
+ 8
6
@@ -18,13 +18,15 @@ logging.basicConfig(
@@ -18,13 +18,15 @@ logging.basicConfig(
logger = logging.getLogger("main")
logger = logging.getLogger("main")
def main(args: argparse.Namespace):
def main():
"""Use CLI arguments to fragment sequences and output text file \
"""Use CLI arguments to fragment sequences and output text file \
with selected terminal fragments.
with selected terminal fragments.
Args:
Args:
args (parser): list of arguments from CLI.
args (parser): list of arguments from CLI.
"""
"""
 
args = parse_arguments()
 
if not isinstance(args, argparse.Namespace):
if not isinstance(args, argparse.Namespace):
raise TypeError("Input should be argparse.Namespace")
raise TypeError("Input should be argparse.Namespace")
@@ -38,8 +40,10 @@ def main(args: argparse.Namespace):
@@ -38,8 +40,10 @@ def main(args: argparse.Namespace):
logger.info("Fragmentation of %s...", args.fasta)
logger.info("Fragmentation of %s...", args.fasta)
splits = np.arange(0, len(list(fasta))+args.size, args.size)
splits = np.arange(0, len(list(fasta))+args.size, args.size)
for i, split in enumerate(splits):
for i in range(len(splits) - 1):
fasta_dict = fasta[split:splits[i+1]]
split = splits[i]
 
keys = list(fasta.keys())[split:splits[i+1]]
 
fasta_dict = {key: fasta[key] for key in keys}
term_frags = fragmentation(fasta_dict, seq_counts,
term_frags = fragmentation(fasta_dict, seq_counts,
args.mean, args.std)
args.mean, args.std)
@@ -132,6 +136,4 @@ if __name__ == '__main__':
@@ -132,6 +136,4 @@ if __name__ == '__main__':
level=logging.INFO,
level=logging.INFO,
)
)
logger = logging.getLogger(__name__)
logger = logging.getLogger(__name__)
main()
arguments = parse_arguments()
main(arguments)
Loading