summaryrefslogtreecommitdiff
path: root/color-converter.py
blob: 785beb8ef35ceb79dd94a25da4bf9e1d98c45bf9 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
# import sys
import argparse

# Order of types must match order of arguments defined
# (or else arg validation will no longer work properly)
TYPES = ['hex', 'rgb', 'cmyk']
# look into CMY, HSL, HSV

HEX_LETTERS = ['a', 'b', 'c', 'd', 'e', 'f']

def main():

    # Set up arguments
    parser = argparse.ArgumentParser()
    
    parser.add_argument('-hex', action='store_true', help='convert from hex (accepts hexadecimal input [ffffff] or string ["#ffffff"] (both are case insensitive)')
    parser.add_argument('-rgb', action='store_true', help='convert from RGB (accepts integer input [R G B], or string [\"rgb(R, G, B)\"] (case/whitespace insensitive)')
    parser.add_argument('-cmyk', action='store_true', help='convert from CMYK (accepts integer input [C M Y K], or string [\"cmyk(C, M, Y, K)\"] (case/whitespace insensitive)')
    
    parser.add_argument('color', nargs='*', help='accepts a color in Hex, RGB, CMYK, or HSL and performs format conversions (does not support CMYK profiles, conversions are uncalibrated)')
    args = parser.parse_args()


    # debug print
    print('args type: ', type(vars(args)))
    print(args, '\n')


    # INPUT SYNTAX VALIDATION
    if not validateArguments(args) :
        return;

    color = args.color
    
    # HANDLE HEX INPUT
    if args.hex :
        color = color[0].lower().strip('#')
        if not validateHex(color) :
            return

        convertFromHex(color)

    # HANDLE RGB INPUT
    if args.rgb :
        # cleanse any non-numerical stuff
        if len(color) == 1 :
            color = color[0].lower().strip('rgb(').strip(')').split(',')
        if not validateRGB(color) :
            return
        
        for i in range(len(color)) :
            color[i] = int(color[i])

        convertFromRGB(color)

    # HANDLE CMYK INPUT
    if args.cmyk :
        # cleanse any non-numerical stuff besides '%'
        if len(color) == 1 :
            color = color[0].lower().strip('cmyk(').strip(')').split(',')
        if not validateCMYK(color) :
            return

        print('convert cmyk: ', color)
##
# HEX CONVERSION SECTION
##

# Takes in valid RGB code and converts it to the other formats
def convertFromHex(hexCode) :
    convertToRGB('hex', hexCode)
    # convertToCMYK('hex', code)
    # convertToHSL('hex', code)

def convertToHex(codeFormat, code) :
    hexValue = '#'
    
    if codeFormat == 'rgb' :
        for rgbValue in code :
            hexValue += hex(int(rgbValue / 16)) 
            hexValue += hex(int(rgbValue % 16))
    
    print('HEX: ', hexValue)


##
# RGB CONVERSION SECTION
##

# Takes in valid RGB code and converts it to the other formats
def convertFromRGB(code) :
    convertToHex('rgb', code)
    # convertToCMYK('rgb', code)
    # convertToHSL('rgb', code)

def convertToRGB(codeFormat, code) :
    rgbValue = '' 

    if codeFormat == 'hex' :
        
        tempSum = 0
        i = 0
        while i < 6 :
            tempSum += int(code[i], 16) * 16
            tempSum += int(code[i+1], 16)
            rgbValue += str(tempSum) + ' '
            i = i + 2
            tempSum = 0
    
    print('RGB: ', rgbValue)



##
# INPUT VALIDATION SECTION
##

# Takes in a string. Returns True if valid Hex color code.
def validateHex(value) :
    if len(value) != 6 :
        print('ERROR: Hex value should have 6 digits')
        return False

    for i in range(len(value)):
        if value[i-1].isnumeric() :
            continue
        elif HEX_LETTERS.count(value[i-1]) != 0 :
            continue
        else :
            print('ERROR: Invalid character in hex code')
            return False

    return True


# Takes in a list of numerical strings. Returns True if valid RGB values. 
def validateRGB(values) :
    if len(values) != 3 :
        print('ERROR: Improper number of values (should be 3)')
        return False

    for value in values :
        if not value.strip().isnumeric() :
            print('ERROR: Improper format for RGB value(s)')
            return False
        value = int(value)
        if (value < 0) or (value > 255) :
            print('ERROR: Each RBG value must be between 0-255')
            return False
    return True

# Takes in a string. Returns True if valid CMYK values.
def validateCMYK(values) :
    if len(values) != 4 :
        print('ERROR: Improper number of values (should be 4)')
        return False
    
    for value in values :
        if not value.strip().strip('%').isnumeric() :
            print('ERROR: Improper format for CMYK value(s). All values must be numeric and between 0-100(%)!')
            return False
        value = int(value)
        if (value < 0) or (value > 100) :
            print('ERROR: Each CMYK value must be between 0-100(%)')
            return False
    
    return True

   

##
# GENERAL UTILITIES
##

# Takes in the program's arguments generated by argparse. Returns True if valid arguments
def validateArguments(args) :
    # First, check that only one input flag is being used
    flagsActive = 0
    for flag in range(len(vars(args))-1) :
        # this uses our TYPES list to key into the args dictionary and determine how many flags are True
        if vars(args)[TYPES[flag]] :
            flagsActive += 1
        if flagsActive > 1 :
            print('ERROR: Currently this tool only supports one input type at a time. Too many flags!')
            return False

    # Second, check that there is a color code to convert
    if len(args.color) == 0 :
        print('ERROR: Must enter both an input flag and color code (did you forget to wrap color code in quotes?)\nFor more info, use the \'-h\' or \'--help\' flag.')
        return False
    
    return True
 

# Takes in a decimal number and converts it to hexadecimal
def hex(number) :
    number = int(number)
    if number > 16 :
        print("ERROR: Decimal to Hexidecimal conversion failed")
        return "ERROR: Decimal to Hexidecimal conversion failed" 
    if number < 10 :
        return str(number)
    return HEX_LETTERS[number % 10]



if __name__ == '__main__' :
    main()