refact: clean up imports

equinor · Nov 16, 2023 · dc95790 · dc95790
1 parent 87389fb
commit dc95790
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/neqsim/__init__.py b/neqsim/__init__.py
@@ -4,7 +4,7 @@
 It uses the Jpype module for bridging python and Java.
 """
 
-from .neqsimpython import *
+from neqsim.neqsimpython import jNeqSim, jpype
 
 def methods(checkClass):
     methods = checkClass.getClass().getMethods()
@@ -23,22 +23,20 @@ def has_tabulate():
 
 
 def setDatabase(connectionString):
-    from neqsim.neqsimpython import jNeqSim
     jNeqSim.util.database.NeqSimDataBase.setConnectionString(connectionString)
     jNeqSim.util.database.NeqSimDataBase.setCreateTemporaryTables(True)
 
 def save_xml(javaobject, filename):
     xstream = jpype.JPackage('com.thoughtworks.xstream')
     streamer = xstream.XStream()
     xml = streamer.toXML(javaobject)
-    print(xml,  file=open(filename, 'w'))
+    print(xml, file=open(filename, 'w'))
     return xml
 
 def open_xml(filename):
     xstream = jpype.JPackage('com.thoughtworks.xstream')
-    security = jpype.JPackage('com.thoughtworks.xstream.security')
     streamer = xstream.XStream()
-    streamer.addPermission(security.AnyTypePermission.ANY)
+    streamer.addPermission(xstream.security.AnyTypePermission.ANY)
     str = open(filename, 'r').read()
     neqsimobj = streamer.fromXML(str)
     return neqsimobj