Home | History | Annotate | Download | only in core 
      1 #!/usr/bin/python -E
      2 
      3 import sys, os, re
      4 
      5 excludes = [r'.*?/\.obj.*?',
      6             r'.*?~',
      7             r'.*?\/.DS_Store',
      8             r'.*?\/.gdb_history',
      9             r'.*?\/buildspec.mk',
     10             r'.*?/\..*?\.swp',
     11             r'.*?/out/.*?',
     12             r'.*?/install/.*?']
     13 
     14 excludes_compiled = map(re.compile, excludes)
     15 
     16 def filter_excludes(str):
     17     for e in excludes_compiled:
     18         if e.match(str):
     19             return False
     20     return True
     21 
     22 def split_perforce_parts(s):
     23     spaces = ((s.count(" ") + 1) / 3) * 2
     24     pos = 0
     25     while spaces > 0:
     26         pos = s.find(" ", pos) + 1
     27         spaces = spaces - 1
     28     return s[pos:]
     29 
     30 def quotate(s):
     31     return '"' + s + '"'
     32 
     33 class PerforceError(Exception):
     34     def __init__(self,value):
     35         self.value = value
     36     def __str__(self):
     37         return repr(self.value)
     38     
     39 
     40 def run(command, regex, filt):
     41     def matchit(s):
     42         m = regex_compiled.match(s)
     43         if m:
     44             return m.group(1)
     45         else:
     46             return ""
     47     def filterit(s):
     48         if filt_compiled.match(s):
     49             return True
     50         else:
     51             return False
     52 
     53     fd = os.popen(command);
     54     lines = fd.readlines()
     55     status = fd.close()
     56     if status:
     57         raise PerforceError("error calling " + command)
     58         
     59     regex_compiled = re.compile(regex)
     60     filt_compiled = re.compile(filt)
     61 
     62     if len(lines) >= 1:
     63         lines = filter(filterit, lines)
     64         if len(lines) >= 1:
     65             return map(matchit, lines)
     66     return None
     67 
     68 try:
     69     if len(sys.argv) == 1:
     70         do_exclude = True
     71     elif len(sys.argv) == 2 and sys.argv[1] == "-a":
     72         do_exclude = False
     73     else:
     74         print "usage: checktree [-a]"
     75         print "  -a  don't filter common crud in the tree"
     76         sys.exit(1)
     77 
     78     have = run("p4 have ...", r'[^#]+#[0-9]+ - (.*)', r'.*')
     79 
     80     cwd = os.getcwd()
     81     files = run("find . -not -type d", r'.(.*)', r'.*')
     82     files = map(lambda s: cwd+s, files)
     83 
     84     added_depot_path = run("p4 opened ...", r'([^#]+)#.*', r'.*?#[0-9]+ - add .*');
     85     added = []
     86     if added_depot_path:
     87         added_depot_path = map(quotate, added_depot_path)
     88 
     89         where = "p4 where " + " ".join(added_depot_path)
     90         added = run(where, r'(.*)', r'.*')
     91         added = map(split_perforce_parts, added)
     92 
     93     extras = []
     94 
     95     # Python 2.3 -- still default on Mac OS X -- does not have set()
     96     # Make dict's here to support the "in" operations below
     97     have = dict().fromkeys(have, 1)
     98     added = dict().fromkeys(added, 1)
     99 
    100     for file in files:
    101         if not file in have:
    102             if not file in added:
    103                 extras.append(file)
    104 
    105     if do_exclude:
    106         extras = filter(filter_excludes, extras)
    107 
    108     for s in extras:
    109         print s.replace(" ", "\\ ")
    110 
    111 except PerforceError, e:
    112     sys.exit(2)
    113 
    114