# -*- coding: utf-8 -*-

import psycopg2
import psycopg2.extras

sql = "select * from accounts"


try:
    conn = psycopg2.connect("dbname='rebate' user='postgresadmin'" \
                            " host='pg-l.cuboidlogic.com' password='P3sdfi334JKd3JxmopfjLIVE1'")
except psycopg2.Error as err:
    print("Connection error: {}".format(err))

try:
    cur = conn.cursor(cursor_factory=psycopg2.extras.RealDictCursor)
    cur.execute(sql)
    data = cur.fetchall()
except psycopg2.Error as err:
    print("Query error: {}".format(err))


data_dict = []
for row in data:
    data_dict.append(dict(row))


cLvlAccs = [0]
cLvl = 1
while len(cLvlAccs) > 0:
    newLvlAccs = []
    for acc in data_dict:
        if (cLvlAccs.count(acc.parent_id) > 0):
            acc.lvl = cLvl;
            newLvlAccs.append(acc.id)
    cLvlAccs = newLvlAccs
    cLvl += 1

for acc in data_dict:
    print "{0} - {1}".format(acc.id, acc.lvl)