renamed to .py so notepad++ users can have syntax highlighting

2011-11-09 13:31:59 +00:00 · 2011-11-09 13:31:59 +00:00 · 8e34309b6e
parent bfd6d4831f
commit 8e34309b6e
1 changed files with 89 additions and 0 deletions
--- a/installation/setup_shellrc.py
+++ b/installation/setup_shellrc.py
@ -0,0 +1,89 @@
+#!/usr/bin/env python
+
+import os,sys,string,re
+from optparse import OptionParser
+
+pathInfo = {\
+            'acml': 'lib/acml4.4.0',
+            'fftw': 'lib/fftw',
+            'msc':  '/msc',
+           }
+
+validShells = {\
+               'bash':['.bashrc','.bash_profile'],
+               'csh': ['.cshrc'],
+              }
+
+environment = {\
+                'LD_LIBRARY_PATH': {'delete':'"acml"',                                                 # what keywords trigger item deletion from existing path 
+                                    'substitute':'[os.path.join(pathInfo["acml"],"ifort64_mp/lib"),\
+                                                   os.path.join(pathInfo["acml"],"ifort64/lib")]',     # what to substitute for deleted path items
+                                    'append': True,                                                    # whether new entries append to existing ${env}
+                                   },
+                'PYTHONPATH':      {'delete':'os.path.join(DamaskRoot,"lib")',
+                                    'substitute':'[os.path.join(DamaskRoot,"lib")]',
+                                    'append': True,
+                                   },
+                'DAMASK_ROOT':     {'delete':'DamaskRoot',
+                                    'substitute':'[DamaskRoot]',
+                                    'append': False,
+                                   },
+              }
+parser = OptionParser(usage="%prog [options]", description = """
+Sets up your shell resource to be compatible with DAMASK. 
+""" + string.replace('$Id$','\n','\\n')
+)
+
+parser.add_option("-s","--shell", type="string", 
+                  dest = "shell", \
+                  help = "type of shell, e.g. "+', '.join(validShells.keys())+" [%default]")
+
+parser.set_defaults(shell = 'bash')
+
+(options, args) = parser.parse_args()
+
+DamaskRoot = os.path.normpath(os.path.join(os.path.dirname(os.path.realpath(sys.argv[0])),'../'))
+
+try:                                        # check for user-defined pathinfo
+  file = open(os.path.join(DamaskRoot,'lib/pathinfo'))
+  content = file.readlines()
+  file.close()
+  for line in content:
+    pathinfo[line.split()[0].lower()] = os.path.normpath(line.split()[1])
+except:
+  pass
+
+theShell = options.shell.lower()
+theHome = os.getenv('USERPROFILE') or os.getenv('HOME')
+
+if theShell == 'bash':
+  for theRC in validShells[theShell]:
+    thePath = os.path.join(theHome,theRC)
+    if os.path.exists(thePath):
+      rc = open(os.path.join(theHome,theRC))
+      content = map(string.strip,rc.readlines())
+      rc.close()
+
+      output = []
+      for envVar in environment.keys(): environment[envVar]['matched'] = False
+  
+      for line in content:
+        for envVar,data in environment.items():
+          m = re.search(r'^(.*?%s=)([^;]*)(.*)$'%envVar,line)
+          if m:
+            substitute = eval(data['substitute'])+[path for path in m.group(2).split(':') if eval(data['delete']) not in path]
+            line = m.group(1)+':'.join(substitute)+m.group(3)
+            environment[envVar]['matched'] = True
+
+        output.append(line)
+
+      for envVar,data in environment.items():
+        if not data['matched']:
+          output.append('export %s=%s'%(envVar,':'.join(eval(data['substitute'])+{True:['${%s}'%envVar],False:[]}[data['append']])))
+
+      rc = open(os.path.join(theHome,theRC),'w')
+      rc.write('\n'.join(output)+'\n')
+      rc.close()
+
+elif theShell == 'csh':
+  print 'csh not supported yet...'