Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- # -*- coding: utf-8 -*-
- import psycopg2
- import psycopg2.extras
- sql = "select * from accounts"
- try:
- conn = psycopg2.connect("dbname='хххххххххх' user='ххххххххххх'" \
- " host='хххххххх' password='ххххххххххх'")
- 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))
- class Account:
- id = 0
- parent_id = 0
- lvl = 0
- def __init__(self, i, p):
- self.id = i
- self.parent_id = p
- 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)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement