1import os 2import sys 3 4from .DominantColors import DominantColors 5 6dir = sys.argv[1] 7m = {} 8 9for f in sorted(os.listdir(dir)): 10 if f.lower().endswith(".jpg") or f.lower().endswith(".jpg"): 11 try: 12 d = DominantColors(os.path.join(dir, f)) 13 calc = d.get_dominant_colors() 14 print(f, "light:", calc[2]) 15 continue 16 17 for fuzzy in range(20): 18 if DominantColors.contains_color(calc, (255, 217, 100), fuzzy): 19 m[f] = fuzzy 20 print(f, fuzzy) 21 break 22 else: 23 m[f] = -1 24 print(f, "no match") 25 26 except Exception: 27 print("oops for " + f) 28 raise 29 # pass 30 31# print "\n----results----" 32# for fuzzy in xrange(10): 33# print fuzzy 34# for k, v in m.items(): 35# if v == fuzzy: 36# print k 37# 38# print "no match:" 39# for k, v in m.items(): 40# if v == -1: 41# print k 42