mmagnus / rna-tools

cmd.extend('edges', edges)

def fr(selection):

    """Save selection into a file in a temp folder and run rna_draw_edges.py on it and load it into this session"""

    my_view = cmd.get_view()

    f = tempfile.TemporaryDirectory()

    tmpf = f.name + os.sep + strip_selection_name(selection) + '.pdb'

    outf = f.name + '/output.py'

    cmd.save(tmpf, selection)

    path = "/Users/magnus/work/src/rna-tools/rna_tools/tools/PyMOL4RNA/"

    cmdline = path + 'rna_draw_frames.py --name %s %s > %s' % (strip_selection_name(selection), tmpf, outf)

    print(cmdline)

    out, err = exe(cmdline)

    if err:

        print(err)

    cmd.load(outf)

    cmd.set_view(my_view)

cmd.extend('fr', fr)

    return selection_name.replace('(', '').replace(')', '')

def edges(selection):

    """Save selection into a file in a temp folder and run rna_draw_edges.py on it and load it into this session"""

    my_view = cmd.get_view()

    f = tempfile.TemporaryDirectory()

    tmpf = f.name + os.sep + strip_selection_name(selection) + '.pdb'

    outf = f.name + '/output.py'

    cmd.save(tmpf, selection)

    cmdline = '/Users/magnus/miniconda3/bin/rna_draw_edges.py --name %s %s > %s' % (strip_selection_name(selection), tmpf, outf)

    print(cmdline)

    out, err = exe(cmdline)

    if err:

        print(err)

    cmd.load(outf)

    cmd.set_view(my_view)

cmd.extend('edges', edges)