Skip to content
Snippets Groups Projects
Verified Commit 3a878b65 authored by Laurent Modolo's avatar Laurent Modolo
Browse files

t2g.py: fix indentation

parent 45be888a
No related branches found
No related tags found
No related merge requests found
......@@ -10,14 +10,32 @@ def validate_file(f):
raise argparse.ArgumentTypeError("{0} does not exist".format(f))
return f
if __name__ == "__main__":
parser = argparse.ArgumentParser(description="create transcript to genes file from a gtf file.")
parser.add_argument("-g", "--gtf", dest="gtf", required=True, type=validate_file,
help="gtf file", metavar="FILE")
args = parser.parse_args()
parser = argparse.ArgumentParser(
description="create transcript to genes file from a gtf file."
)
parser.add_argument(
"-g", "--gtf", dest="gtf", required=True, type=validate_file,
help="gtf file", metavar="FILE"
)
args = parser.parse_args()
db = gffutils.create_db(args.gtf, dbfn = ":memory:", force = True, merge_strategy="merge", disable_infer_transcripts=True, disable_infer_genes=True)
with open("t2g.txt", "w") as t2g:
for gene in db.all_features():
for transcript in db.children(gene, featuretype='transcript', order_by='start'):
t2g.write(str(transcript["transcript_id"][0]) + "\t" + str(gene["gene_id"][0]) + "\n")
db = gffutils.create_db(
args.gtf,
dbfn=":memory:",
force=True,
merge_strategy="merge",
disable_infer_transcripts=True,
disable_infer_genes=True
)
with open("t2g.txt", "w") as t2g:
for gene in db.all_features():
for transcript in db.children(
gene, featuretype='transcript', order_by='start'):
t2g.write(
str(transcript["transcript_id"][0]) +
"\t" +
str(gene["gene_id"][0]) +
"\n"
)
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment