From d95826276180671a57454d09fb5610ea5ceaf384 Mon Sep 17 00:00:00 2001
From: Anand Doshi <anand@iwebnotes.com>
Date: Tue, 5 Jul 2011 11:42:24 +0530
Subject: [PATCH] Added method is_setup_okay to confirm if a db is setup
 properly

---
 setup/doctype/setup_control/setup_control.py | 47 ++++++++++++++------
 1 file changed, 34 insertions(+), 13 deletions(-)

diff --git a/setup/doctype/setup_control/setup_control.py b/setup/doctype/setup_control/setup_control.py
index 284e2a8a36..7cbf4d32fb 100644
--- a/setup/doctype/setup_control/setup_control.py
+++ b/setup/doctype/setup_control/setup_control.py
@@ -13,6 +13,8 @@ sql = webnotes.conn.sql
 get_value = webnotes.conn.get_value
 in_transaction = webnotes.conn.in_transaction
 convert_to_lists = webnotes.conn.convert_to_lists
+
+from server_tools.gateway_utils import update_client_control, get_total_users
 	
 # -----------------------------------------------------------------------------------------
 
@@ -36,10 +38,10 @@ class DocType:
 	#-----------------------
 	def set_account_details(self, args):
 		args = eval(args)
-		
-		self.set_cp_defaults(args['company_name'], args['industry'], args['time_zone'], args['country'], args['account_name'])
+		#webnotes.logger.error("args in set_account_details of setup_control: " + str(args))
+		self.set_cp_defaults(args['company'], args['industry'], args['time_zone'], args['country'], args['account_name'])
 		self.create_profile(args['user'], args['first_name'], args['last_name'])	
-		self.update_client_control()
+		update_client_control(args['total_users'])
 		
 	
 	# Account Setup
@@ -163,19 +165,38 @@ class DocType:
 			d = addchild(pr,'userroles', 'UserRole', 1)
 			d.role = r
 			d.save(1)
-	
-	# Update WN ERP Client Control
-	# -----------------------------
-	def update_client_control(self):
-		cl = Document('WN ERP Client Control','WN ERP Client Control')
-		cl.account_start_date = nowdate()
-		cl.total_users = 1
-		cl.is_trial_account = 1
-		cl.save()
+			
 
 	# Sync DB
 	# -------
 	def sync_db(arg=''):
 		import webnotes.model.db_schema
 		sql("delete from `tabDocType Update Register`")
-		webnotes.model.db_schema.sync_all()
\ No newline at end of file
+		webnotes.model.db_schema.sync_all()
+		
+	
+	def is_setup_okay(self, args):
+		"""
+		Validates if setup has been performed after database allocation
+		"""
+		
+		args = eval(args)
+		#webnotes.logger.error("args in set_account_details of setup_control: " + str(args))
+		
+		cp_defaults = webnotes.conn.get_value('Control Panel', None, 'account_id')
+		
+		user_profile = webnotes.conn.get_value('Profile', args['user'], 'name')
+		
+		from webnotes.utils import cint
+		
+		total_users = get_total_users()
+		
+		#webnotes.logger.error("setup_control.is_setup_okay: " + cp_defaults + " " + user_profile + " " + str(total_users))
+		
+		#webnotes.logger.error("setup_control.is_setup_okay: Passed Values:" + args['account_name'] + " " + args['user'] + " " + str(args['total_users']))
+		
+		
+		if (cp_defaults==args['account_name']) and user_profile and \
+		   (total_users==cint(args['total_users'])):
+		   return 'True'
+		
\ No newline at end of file