#!/usr/bin/python
import os, psycopg2, sys, xmlrpclib, string, time
from threading import Thread


dbname = sys.stdin.read().strip() #Replace with targeted openerp's database
dbuser = 'bino'
erpuser = 'admin'
erppasswd = 'admin'

def myfunc(ctr_code,ctr_name,db_con,erp_con):
   ctr_data = {
      'code': ctr_code,
      'name': ctr_name
   }
   ctrid = erp_con.execute(dbname, dbuid, erppasswd , 'pbx.country', 'create', ctr_data)
   #con_a = psycopg2.connect("dbname=%(vdbname)s user=%(vdbuser)s" %con_dict)
   cur_a = db_con.cursor()
   cur_a.execute("SELECT code,name FROM garea WHERE ccode =  %(xccode)s", dict(xccode=ctr_code))
   for area in cur_a.fetchall():
      area_code = area[0]
      area_name = area[1]
      if area[0] is None: #No Area code
         area_name = ctr_name +" (All Area)" #Means its the hole country
      area_data = {
         'ccode': ctrid,
         'code': area_code,
         'name': area_name
      }
      area_id = erp_con.execute(dbname, dbuid, erppasswd, 'pbx.area', 'create', area_data)
   cur_a.close()

#Tables for source
os.system("psql -d %s -U %s -f './prefx.sql'" % (dbname,dbuser))

#get UID
sock = xmlrpclib.ServerProxy('http://localhost:8069/xmlrpc/common',allow_none=1)
dbuid = sock.login(dbname, erpuser, erppasswd)
sock = xmlrpclib.ServerProxy('http://localhost:8069/xmlrpc/object',allow_none=1)

#("host=%s dbname=%s user=%s password=%s" % (DBHOST,DBDATABASE, DBUSER, DBPASSWORD))
#con_c = psycopg2.connect("dbname=%(v_dbname)s user=%(v_dbuser)s", serialize=0, dict(v_dbname=dbname v_dbuser=dbuser))
con_dict = {'vdbname': dbname , 'vdbuser': dbuser}
con_c = psycopg2.connect("dbname=%(vdbname)s user=%(vdbuser)s" %con_dict)
cur_c = con_c.cursor()
cur_c.execute("SELECT * FROM genctr")
#prepare multi connection
con_pool = []
erp_pool = []
for i in range(1, 7):
   con_pool.append(psycopg2.connect("dbname=%(vdbname)s user=%(vdbuser)s" %con_dict))
   erp_pool.append(xmlrpclib.ServerProxy('http://localhost:8069/xmlrpc/object',allow_none=1))
con_num = 0
for ctr in cur_c.fetchall():
   #Add one Country to pbx_country
   ctr_code = ctr[0]
   ctr_name = ctr[1]
   print ctr_code + " | " +ctr_name
   if con_num > 5:
      con_num = 0

   t = Thread(target=myfunc, args=(ctr_code,ctr_name,con_pool[con_num], erp_pool[con_num])) #Struct the threads
   t.start() # Start the thread

   con_num += 1
cur_c.close()

bino@erp:~/mydoc/openerp/smdr$ echo pbx01 |./uldata.py