create__init__.py 860 B

12345678910111213141516171819202122232425262728293031323334
  1. #!/usr/bin/env python
  2. import os
  3. import sys
  4. import glob
  5. def main(path):
  6. if not os.path.exists(path) or not os.path.isdir(path):
  7. print >> sys.stderr, "'%s' is not a directory" % path
  8. return 1
  9. modules = []
  10. for f in glob.glob(os.path.join(os.path.basename(path), '*.py')):
  11. if f[-5:-3] == '__':
  12. continue
  13. modules.append(os.path.splitext(os.path.basename(f))[0])
  14. fd = open(os.path.join(path, '__init__.py'), 'w')
  15. try:
  16. fd.write('all = [%s' % os.linesep)
  17. for m in modules:
  18. fd.write(" '%s',%s" % (m, os.linesep))
  19. fd.write(' ]%s' % os.linesep)
  20. finally:
  21. fd.close()
  22. return 0
  23. if __name__ == "__main__":
  24. if len(sys.argv) < 2:
  25. sys.exit("usage: %s path/to/gui_modules" % sys.argv[0])
  26. sys.exit(main(sys.argv[1]))