From 3f614bc7b1dd5c95628f4c448b9d8a4ae6e8b1cd Mon Sep 17 00:00:00 2001
From: mpa <mpa@opencascade.com>
Date: Tue, 23 Sep 2014 13:57:32 +0400
Subject: [PATCH] Edited python commands for single-study

---
 src/KERNEL_PY/__init__.py     | 13 +++++++++
 src/KERNEL_PY/salome_study.py | 50 +++++++++++++----------------------
 2 files changed, 31 insertions(+), 32 deletions(-)

diff --git a/src/KERNEL_PY/__init__.py b/src/KERNEL_PY/__init__.py
index 6933a14f2..7cebb6be8 100755
--- a/src/KERNEL_PY/__init__.py
+++ b/src/KERNEL_PY/__init__.py
@@ -216,6 +216,7 @@ def salome_init(theStudyId=0,embedded=0):
     global sg
     global myStudyManager, myStudyId, myStudy, myStudyName
 
+    print "*** import salome ***"
     try:
         if salome_initial:
             salome_initial=0
@@ -238,13 +239,25 @@ def salome_init(theStudyId=0,embedded=0):
         ============================================
         """
         raise
+    
+def study_close():
+    print "*** study_close ***"
+    if myStudy.IsEmpty():
+        raise RuntimeError, "Study is already closed!"
+    else:
+        myStudy.Close()
+        salome_close()
+        pass 
+    pass
 
 def salome_close():
+    print "*** salome_close ***"
     global salome_initial
     salome_initial=1
     salome_iapp_close()
     salome_kernel_close()
     salome_study_close()
+    myStudyId, myStudy, myStudyName=None,None,None
     pass
 
 
diff --git a/src/KERNEL_PY/salome_study.py b/src/KERNEL_PY/salome_study.py
index 29ba67d03..cbe7106cd 100755
--- a/src/KERNEL_PY/salome_study.py
+++ b/src/KERNEL_PY/salome_study.py
@@ -271,29 +271,21 @@ salome_study_ID = -1
 
 def getActiveStudy(theStudyId=0):
     global salome_study_ID
+    if salome_study_initial:
+        print "No active study"
+        return None
+        pass
 
     if verbose(): print "getActiveStudy"
     if salome_study_ID == -1:
-        if salome_iapp.hasDesktop():
-            if verbose(): print "---in gui"
-            salome_study_ID = salome_iapp.sg.getActiveStudyId()
+        listOpenStudies = myStudyManager.GetOpenStudies()
+        if len(listOpenStudies) == 0:
+            return None
         else:
-            if verbose(): print "---outside gui"
-            if theStudyId:
-                aStudy=myStudyManager.GetStudyByID(theStudyId)
-                if aStudy:
-                    if verbose(): print "connection to existing study ", theStudyId
-                    salome_study_ID = theStudyId
-            if salome_study_ID == -1:
-                listOpenStudies = myStudyManager.GetOpenStudies()
-                if len(listOpenStudies) == 0:
-                    salome_study_ID = createNewStudy()
-                else:
-                    s = myStudyManager.GetStudyByName(listOpenStudies[0])
-                    salome_study_ID = s._get_StudyId()
-            else:
-                pass
-            if verbose(): print"--- Study Id ", salome_study_ID
+            print "getActiveStudy: GetStudyByName, salome_study_ID = ",salome_study_ID
+            s = myStudyManager.GetStudyByName(listOpenStudies[0])
+            salome_study_ID = s._get_StudyId()
+    if verbose(): print"--- Study Id ", salome_study_ID
     return salome_study_ID
 
     #--------------------------------------------------------------------------
@@ -335,18 +327,7 @@ def setCurrentStudyId(theStudyId=0):
 
 def createNewStudy():
     print "createNewStudy"
-    i=1
-    aStudyName = "noName"
-    nameAlreadyInUse = 1
-    listOfOpenStudies = myStudyManager.GetOpenStudies()
-    print listOfOpenStudies
-    while nameAlreadyInUse:
-        aStudyName = "extStudy_%d"%i
-        if aStudyName not in listOfOpenStudies:
-            nameAlreadyInUse=0
-        else:
-            i = i+1
-
+    aStudyName = "extStudy"
     theStudy = myStudyManager.NewStudy(aStudyName)
     theStudyId = theStudy._get_StudyId()
     print aStudyName, theStudyId
@@ -366,7 +347,7 @@ def salome_study_init(theStudyId=0):
         n (>0) : try connection to study with Id = n, or create a new one
                  if study not found.
     """
-
+    print "*** salome_study_init ***"
     global salome_study_initial
     global myStudyManager, myStudyId, myStudy, myStudyName
     global orb, lcc, naming_service, cm
@@ -384,6 +365,9 @@ def salome_study_init(theStudyId=0):
 
         # get active study Id, ref and name
         myStudyId = getActiveStudy(theStudyId)
+        if myStudyId == None :
+            myStudyId = createNewStudy()
+            print "getActiveStudy: createNewStudy, salome_study_ID = ",salome_study_ID
         if verbose(): print "myStudyId",myStudyId
         myStudy = myStudyManager.GetStudyByID(myStudyId)
         myStudyName = myStudy._get_Name()
@@ -391,7 +375,9 @@ def salome_study_init(theStudyId=0):
     return myStudyManager, myStudyId, myStudy, myStudyName
 
 def salome_study_close():
+    print "*** salome_study_close ***"
     global salome_study_initial, salome_study_ID
     salome_study_initial=1
     salome_study_ID=-1
+    myStudyId, myStudy, myStudyName=None,None,None
     pass
-- 
2.39.2