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