import pymysql.cursors

def get_wc(post_nr):
    #alt denne information skal ændres baseret på hvis pc det køre på og hvad databasen hedder
    cnx = pymysql.connect(user='root', password='1234', host='127.0.0.1', database='iToilet')
    cursor = cnx.cursor()

    to_return = []
    query = ("SELECT zip, adress FROM toilets where zip =" + str(post_nr))
    cursor.execute(query)

    for (zip_, adress) in cursor:
        #to_return.append([zip_, adress])
        to_return.append("{}, {}".format(zip_, adress))

    cursor.close()
    cnx.commit()
    cnx.close()
    return to_return

def create_wc(l):
    #alt denne information skal ændres baseret på hvis pc det køre på og hvad databasen hedder
    cnx = pymysql.connect(user='root', password='1234', host='127.0.0.1', database='iToilet')
    cursor = cnx.cursor()

    for i in l:
        query = ("INSERT INTO toilets (zip, adress) VALUES (%s,%s)")
        query_data = (i[1],i[0])
        # if sætning findes kun for at fange toiletter uden post nr.
        if i[1] != "No results found":
            cursor.execute(query, query_data)

    cursor.close()
    cnx.commit()
    cnx.close()