minor layout changes to script

Author: facu.r

netpyne_ui/netpyne_geppetto.py

@@ -405,23 +405,24 @@ def generateScript(self, metadata):
         def convert2bool(string):
             return string.replace('true', 'True').replace('false', 'False')
 
-        def header(title):
-            sep = (76-len(title))//2
-            return '\n# ' + '-'*sep + ' ' + title.upper() + ' ' + '-'*sep + '\n'
+        def header(title, spacer='-'):
+            return '\n# ' + title.upper() + ' ' + spacer*(77-len(title)) + '\n'
 
         try :
             params =  ['popParams' , 'cellParams', 'synMechParams']
             params += ['connParams', 'stimSourceParams', 'stimTargetParams']
 
-            with open(metadata['scriptName']+'.py', 'w') as script:
+            fname = metadata['scriptName'] if metadata['scriptName'][-3:]=='.py' else metadata['scriptName']+'.py'
+            
+            with open(fname, 'w') as script:
                 script.write('from netpyne import specs, sim\n')
-                script.write(header('documentation '))
+                script.write(header('documentation'))
                 script.write("''' Script generated with NetPyNE-UI. Please visit:\n")
                 script.write("    - https://www.netpyne.org\n    - https://github.com/MetaCell/NetPyNE-UI\n'''\n")
-                script.write(header('script'))
+                script.write(header('script', spacer='='))
                 script.write('netParams = specs.NetParams()\n')
                 script.write('simConfig = specs.SimConfig()\n')
-                script.write(header('single  value attributes'))
+                script.write(header('single value attributes'))
                 for attr, value in netParams.__dict__.items():
                     if attr not in params:
                         if value!=getattr(specs.NetParams(), attr):
@@ -434,7 +435,7 @@ def header(title):
                         script.write("netParams." + param + "['" + key + "'] = ")
                         script.write(convert2bool(json.dumps(value, indent=4))+'\n')
 
-                script.write(header('network  configuration'))
+                script.write(header('network configuration'))
                 for attr, value in simConfig.__dict__.items():
                     if value!=getattr(specs.SimConfig(), attr):
                         script.write('netParams.' + attr + ' = ')
@@ -443,7 +444,7 @@ def header(title):
                 script.write(header('create simulate analyze  network'))
                 script.write('sim.createSimulateAnalyze(netParams=netParams, simConfig=simConfig)\n')
 
-                script.write(header('end script'))
+                script.write(header('end script', spacer='='))
 
             return self.getJSONReply()