Dome/nogui.py

   1 #!/usr/bin/env python
   2
   3 import findrox
   4
   5 no_gui_mode = 1
   6
   7 from Beep import Beep
   8
   9 #import gc
  10 #gc.set_debug(gc.DEBUG_LEAK)
  11
  12 try:
  13         import rox
  14 except:
  15         print "(no GUI, but that's OK)"
  16
  17 from rox import choices, support
  18
  19 def error(message, title = 'Error'):
  20         print "*********", title
  21         print message, "\n"
  22 support.report_error = error
  23
  24 import sys
  25 #from xml.dom import ext
  26 from Ft.Xml.Domlette import PrettyPrint
  27
  28 from Model import Model
  29 from View import View, Done, InProgress
  30 from Program import Program, load_dome_program
  31
  32 if len(sys.argv) > 1 and sys.argv[1] == '--profile':
  33         profiling = 1
  34         del sys.argv[1]
  35 else:
  36         profiling = 0
  37
  38 if len(sys.argv) > 1 and sys.argv[1] == '--xmlonly':
  39         xmlonly = 1
  40         del sys.argv[1]
  41 else:
  42         xmlonly = 0
  43
  44 if len(sys.argv) < 2:
  45         print "Usage: python nogui.py [--profile] [--xmlonly] <document>"
  46         sys.exit(0)
  47
  48 idle_list = []
  49
  50 class Callback:
  51         def __init__(self, fn):
  52                 self.fn = fn
  53
  54 def idle_add(function):
  55         new = Callback(function)
  56         idle_list.append(new)
  57         return new
  58
  59 def idle_remove(tag):
  60         try:
  61                 idle_list.remove(tag)
  62         except ValueError:
  63                 pass
  64
  65 source = sys.argv[1]
  66 if len(sys.argv) > 2:
  67         xml_data = sys.argv[2]
  68 else:
  69         xml_data = None
  70 model = Model(source, dome_data = xml_data)
  71
  72 view = View(model, callback_handlers = (idle_add, idle_remove))
  73
  74 def run_nogui():
  75         print "Starting root program of", source
  76
  77         view.run_new(None)
  78
  79         try:
  80                 view.do_action(['play', model.root_program.name])
  81         except InProgress:
  82                 pass
  83         except Beep:
  84                 print "*** BEEP ***"
  85                 raise
  86
  87         while idle_list:
  88                 for i in idle_list[:]:
  89                         if not i.fn():
  90                                 idle_remove(i)
  91
  92         print "Done!"
  93
  94 if profiling:
  95         import profile
  96         print "Profiling..."
  97         profile.run('run_nogui()')
  98 else:
  99         run_nogui()
 100
 101 if view.chroots:
 102         raise Exception("Processing stopped in a chroot! -- not saving")
 103
 104 view.model.strip_space()
 105
 106 import shutil
 107
 108 if not xmlonly:
 109         shutil.copyfile(source, source + '.bak')
 110         doc = view.export_all()
 111         PrettyPrint(doc, stream = open(source, 'w'))
 112
 113 if xml_data:
 114         output = xml_data
 115 else:
 116         if source[-5:] == '.dome':
 117                 output = source[:-5] + '.xml'
 118         else:
 119                 output = source + '.xml'
 120 PrettyPrint(view.model.doc, stream = open(output + '.new', 'w'))
 121
 122 import os
 123 os.rename(output + '.new', output)