fileslist.py 401 B

12345678910111213141516171819202122
  1. #!/usr/bin/env python
  2. # -*- coding: utf-8 -*-
  3. import os
  4. import sys
  5. path = sys.argv[1]
  6. prefix = sys.argv[2]
  7. ls = os.listdir(path)
  8. ext = [".c", ".cpp", ".h"]
  9. fn = open("ls.txt", "w")
  10. def write(*ext):
  11. for name in ls:
  12. e = os.path.splitext(name)[1]
  13. if e in ext:
  14. nm = "\t" + prefix + "/" + name
  15. fn.write(nm + " \ \n")
  16. write(".c", ".cpp")
  17. write(".h", ".hpp")