def fetch_door_id(self):
serial_num = get_serial()
if serial_num is None:
raise Exception("Unable to get serial number!")
c = self.conn.cursor()
c.execute("SELECT id FROM records_door WHERE serial = '%s'" % serial_num)
result = c.fetchone()
if result:
return result[0]
else:
raise Exception("Register this door with ForkPi first!")
def fetch_door_id(self):
serial_num = get_serial()
if serial_num is None:
raise Exception("Unable to get serial number!")
c = self.conn.cursor()
c.execute("SELECT id FROM records_door WHERE serial = '%s'" %
serial_num)
result = c.fetchone()
if result:
return result[0]
else:
raise Exception('Register this door with ForkPi first!')
conn = psycopg2.connect(database="forkpi", user="******", password="******", host="forkpi.local", port="5432")
c = conn.cursor()
c.execute("SELECT id FROM records_door WHERE name='%s' AND serial != '%s'"% (door_name, serial_num))
result = c.fetchone()
if result:
print("The name '%s' is already taken. Please choose another name." % door_name)
else:
c.execute("SELECT id, name FROM records_door WHERE serial='%s'" % serial_num)
result = c.fetchone()
if result:
c.execute("UPDATE records_door SET name='%s' WHERE id=%s" % (door_name, result[0]))
conn.commit()
print("Changed name from '%s' to '%s'!" % (result[1], door_name))
else:
c.execute("INSERT INTO records_door(name, serial) VALUES ('%s', '%s')" % (door_name, serial_num))
conn.commit()
print("Successfully registered as '%s'!" % door_name)
conn.close()
if __name__ == '__main__':
if len(sys.argv) == 1:
print('Usage: python3 %s <door name>' % sys.argv[0])
else:
door_name = ' '.join(sys.argv[1:]).replace('\'', '').replace('\n', '')
serial_num = get_serial()
if serial_num:
register_door(serial_num, door_name)
else:
print('Unable to get serial number!')
