#!/usr/bin/python
# Einfaches Python Kommandozeilen-Tool fuer den MCP23017 I2C IO Expander

import smbus
import sys
import getopt
 
# RasPi Revision 1: bus = smbus.SMBus(0) 
# RasPi Revision 2: bus = smbus.SMBus(1)
bus = smbus.SMBus(1) 

# I2C address of MCP23017
address = 0x20

bus.write_byte_data(address,0x00,0x00) # Set all of bank A to outputs 
bus.write_byte_data(address,0x01,0x00) # Set all of bank B to outputs 

# Let them know how it works
def usage():
   print 'Usage: mcp23017.py -b <bank> -o <output> -s <state>'
   print 'bank ::= a|b; output ::= 0|1|2|3|5|6|7; state ::= 0|1|r;'

# Handle the command line arguments
def main():
   try:
      opts, args = getopt.getopt(sys.argv[1:],"hb:o:s:",["bank=","output=","state="])

      if not opts:
        usage()
        sys.exit(2)

   except getopt.GetoptError:
      usage()
      sys.exit(2)

   for opt, arg in opts:
      if opt == '-h':
         usage()
         sys.exit()
      elif opt in ("-b", "--bank"):
         bank = arg
      elif opt in ("-o", "--output"):
         output = int(arg)
      elif opt in ("-s", "--state"):
         state = arg

# Set the correct register for the banks
   if bank == "a" :
    register = 0x12
   elif bank == "b" :
    register = 0x13
   else:
    print "Error! Bank must be a or b"
    sys.exit()

# Read current values from the IO Expander
   value =  bus.read_byte_data(address,register) 

# Shift the bits for the register value, checking if they are already set first
   if state == "1":
    if (value >> output) & 1 :
     print "Output GP"+bank.upper()+str(output), "is already high."
     sys.exit()
    else:
      value += (1 << output)
   elif state == "0":
    if (value >> output) & 1 :
     value -= (1 << output)
    else:
     print "Output GP"+bank.upper()+str(output), "is already low."
     sys.exit()
   elif state == "r":
    if (value >> output) & 1 :
     print "high"
    else:
     print "low"
    sys.exit()
   else:
    print "Error! state must be r, 1 or 0"
    sys.exit()

# Now write to the IO expander
   bus.write_byte_data(address,register,value)
 
# Tell them what we did
   print "Output GP"+bank.upper()+str(output), "changed to", state

if __name__ == "__main__":
   main()