rgbcsv2colordict.py

#!/usr/bin/env python
# rgbcvs2colorarray.py
# by wilsonmar@gmail.com
# Usage: rgbcvs2colorarray.py  rgb_combined.csv
# Oputput: A = array([[222,43,221],[2,11,222], ... ])
# Open resulting file rgb_colorarray.py and paste into rgb2colorname.py

from __future__ import print_function

# Begin timer:
import timeit
start_time = timeit.default_timer()

# Get argument list using sys module:
import sys, os.path
def program(*args):
    # do whatever
    pass
# Provide default file_in name argument if not provided:
if __name__ == "__main__":
#def main(argv):
   try:
      arg1 = sys.argv[1]
      file_in = sys.argv[1]
   except IndexError: # getopt.GetoptError:
      # print "Usage: " + sys.argv[0] + ' -i <inputfile> -o <outputfile>'
      # sys.exit(2)
      file_in = 'rgb_combined_v05.csv'

# Exit if file_in not found:
if os.path.exists(file_in) and os.access(file_in, os.R_OK):
    import csv, time
    with open(file_in, 'rU') as f:
        reader = csv.reader(f, delimiter=',')
        for i in reader:
            header_rows = '# '+time.strftime('%Y-%m-%d %H:%M (local time)')+" "+sys.argv[0]+' START: inrowcount='+ str( sum(1 for _ in f) ) +'.'
            print(header_rows)
else:
    print('# '+time.strftime('%Y-%m-%d %H:%M (local time)')+' '+sys.argv[0]+" ABORTED. Either file "+file_in+" is missing or is not readable.")
    exit(2)

# Provide default file_out name argument if not provided:
if __name__ == "__main__":
#def main(argv):
   try:
      arg1 = sys.argv[2]
      file_out = sys.argv[2]
   except IndexError: # getopt.GetoptError:
   	  # Name output file by appending .txt to the name:
      file_out = file_in + '.txt'

# Send STDOUT to a file:
stdout = sys.stdout  # remember the handle to the real standard output.
sys.stdout=open( file_out,"w")


# Print in yml format:
import csv
# 'rU' means open in universal-newline mode needed on Macs:
#with open('./Portfolio.csv', 'rU') as f:

with open(file_in, 'rU') as f:
    reader = csv.reader(f, delimiter=',')
    first_line = f.readline() # pull out first line - do not print 
    print("    HexNameDict={",end="") # no NewLine
    # Loop through lines in input to generate: "[222,43,221],[2,11,222]", one for each color:
    rownum=0
    for i in reader:
      strRGBHex=i[0]
      if rownum ==0:
          # print first row without a comma:
          print( '"'+i[0]+'":"'+i[4]+'"',end="")
          lastRGBHex=strRGBHex
          rownum=rownum+1
      else:
        if lastRGBHex == strRGBHex:
           pass
        else:
          print(',"'+i[0]+'":"'+i[4]+'"',end="")
          lastRGBHex=strRGBHex
          rownum=rownum+1
    print("}") # end with NewLine

footer_row1="# "+ time.strftime('%Y-%m-%d %H:%M (local time)') +' '+ sys.argv[0] +' '+ file_out +" output "+ str(rownum)+ ' rows.'
print("    "+footer_row1)
# Close the file every time:
sys.stdout.close()

sys.stdout = stdout # Restore regular stdout.
print(footer_row1)
elapsed = timeit.default_timer() - start_time # End timer:
print("# "+ time.strftime('%Y-%m-%d %H:%M (local time)') +' '+ sys.argv[0] +" END: ran for "+ "{:.2f}".format(elapsed * 1000 )+ ' secs.')