comparison sRbowtieParser.py @ 2:70193ce0540e draft

planemo upload for repository https://bitbucket.org/drosofff/gedtools/

1:f5388910d083

#!/usr/bin/python
# python parser module to analyse sRbowtie alignments
# version 1.0.0 - argparse implementation
# Usage sRbowtieParser.py  <1:index source> <2:extraction directive> <3:outputL> <4:polarity> <5:6:7 filePath:FileExt:FileLabel> <.. ad  lib>

import sys, argparse
from smRtools import *

def Parser():
  the_parser = argparse.ArgumentParser()
  the_parser.add_argument('--IndexSource', action="store", type=str, help="Path to the index source")
  the_parser.add_argument('--ExtractDirective', action="store", type=str, choices=["fastaSource", "bowtieIndex"], help="Extract info from fasta or bowtie index")

IndexSource = args.IndexSource
genomeRefFormat = args.ExtractDirective
Output = args.output
Polarity = args.polarity
MasterListOfGenomes = {}

for filePath, FileExt, FileLabel in zip (args.alignmentSource, args.alignmentFormat, args.alignmentLabel):
  MasterListOfGenomes[FileLabel] = HandleSmRNAwindows (filePath, FileExt, IndexSource, genomeRefFormat) 

header = ["gene"]
for filePath, FileExt, FileLabel in zip (args.alignmentSource, args.alignmentFormat, args.alignmentLabel):
  header.append(FileLabel)

F = open (args.output, "w")
# print >>F, args
print >> F, "\t".join(header)
for item in sorted (MasterListOfGenomes[header[1]].instanceDict.keys() ):
  line=[item]
  for sample in header[1:]:
    count = str (MasterListOfGenomes[sample].instanceDict[item].readcount(polarity=Polarity))
    line.append(count)
  print >> F,  "\t".join(line )
F.close()

2:70193ce0540e

#!/usr/bin/python
# python parser module to analyse sRbowtie alignments
# version 1.0.2 - argparse implementation
# Usage sRbowtieParser.py  <1:index source> <2:extraction directive> <3:outputL> <4:polarity> <5:6:7 filePath:FileExt:FileLabel> <.. ad  lib>

import sys, argparse
from smRtools import *

def masterListGenerator(data_source):
  for filePath, FileExt, FileLabel in data_source:
    yield HandleSmRNAwindows (filePath, FileExt, IndexSource, genomeRefFormat)

def Parser():
  the_parser = argparse.ArgumentParser()
  the_parser.add_argument('--IndexSource', action="store", type=str, help="Path to the index source")
  the_parser.add_argument('--ExtractDirective', action="store", type=str, choices=["fastaSource", "bowtieIndex"], help="Extract info from fasta or bowtie index")

IndexSource = args.IndexSource
genomeRefFormat = args.ExtractDirective
Output = args.output
Polarity = args.polarity
header = ["gene"]


FileLabelList=[label for label in args.alignmentLabel]
header.extend(FileLabelList)
assert (len(FileLabelList)==len(set(FileLabelList))),"You have supplied a non-unique label. Please make sure that your input files have unique names"

data_source=zip (args.alignmentSource, args.alignmentFormat, args.alignmentLabel)
master_generator=masterListGenerator(data_source)

for i,window in enumerate(master_generator):
  window=window
  if i==0:
    gene_count_dict={gene:[str(item.readcount(polarity=Polarity))] for gene,item in window.instanceDict.items()}
  else:
    [gene_count_dict[gene].append(str(item.readcount(polarity=Polarity))) for gene,item in window.instanceDict.items()]


F = open (args.output, "w")
# print >>F, args
print >> F, "\t".join(header)

for item in sorted(gene_count_dict.keys()):
  line=[item]
  line.extend(gene_count_dict[item])
  print >> F,  "\t".join(line )
F.close()