source: branches/hpc34/python/env.py@ 2708

Last change on this file since 2708 was 2499, checked in by Malte Marquarding, 12 years ago

use ipython >=0.11

File size: 3.6 KB
Line 
1"""This module has various functions for environment specific setings.
2"""
3__all__ = ["is_casapy", "is_ipython", "setup_env", "get_revision",
4 "is_asap_cli"]
5
6import sys
7import os
8import shutil
9import platform
10
11def is_casapy():
12 """Are we running inside casapy?"""
13 try:
14 from taskinit import casalog
15 return True
16 except ImportError:
17 return False
18
19def is_ipython():
20 """Are we running inside IPython?"""
21 return 'IPython' in sys.modules.keys()
22
23def is_asap_cli():
24 """Are we running inside asap ipython (but not casapy)"""
25 return is_ipython() and not is_casapy()
26
27def setup_env():
28 """Set-up environment variables for casa and initialise ~/.asap on first
29 use.
30 """
31 # Set up CASAPATH and first time use of asap i.e. ~/.asap/*
32 plf = None
33 if sys.platform == "linux2":
34 if platform.architecture()[0] == '64bit':
35 plf = 'linux_64b'
36 else:
37 plf = 'linux_gnu'
38 elif sys.platform == 'darwin':
39 plf = 'darwin'
40 else:
41 # Shouldn't happen - default to linux
42 plf = 'linux'
43 asapdata = os.path.split(__file__)[0]
44
45 # Allow user defined data location
46 if os.environ.has_key("ASAPDATA"):
47 if os.path.exists(os.environ["ASAPDATA"]):
48 asapdata = os.environ["ASAPDATA"]
49 # use CASAPATH if defined and "data" dir present
50 if not os.environ.has_key("CASAPATH") or \
51 not os.path.exists(os.environ["CASAPATH"].split()[0]+"/data"):
52 os.environ["CASAPATH"] = "%s %s somwhere" % ( asapdata, plf)
53
54 # set up user space
55# userdir = os.environ["HOME"]+"/.asap"
56# if not os.path.exists(userdir):
57# print 'First time ASAP use. Setting up ~/.asap'
58# os.makedirs(os.path.join(userdir, "profile_default"))
59# if not is_casapy():
60# shutil.copyfile(asapdata+"/data/ipy_user_conf.py",
61# userdir+"/ipy_user_conf.py")
62# shutil.copyfile(asapdata+"/data/ipython_config.py",
63# userdir+"/profile_default/ipython_config.py")
64# f = file(userdir+"/asapuserfuncs.py", "w")
65# f.close()
66# f = file(userdir+"/ipythonrc", "w")
67# f.close()
68# else:
69# if not is_casapy():
70# # upgrade to support later ipython versions
71
72# if not os.path.exists(userdir+"/ipy_user_conf.py"):
73# shutil.copyfile(asapdata+"/data/ipy_user_conf.py",
74# userdir+"/ipy_user_conf.py")
75# if not os.path.exists(userdir+"/profile_default/ipython_config.py"):
76# shutil.copyfile(asapdata+"/data/ipython_config.py",
77# userdir+"/profile_default/ipython_config.py")
78
79
80def get_revision():
81 """Get the revision of the software. Only useful within casapy."""
82 if not is_casapy:
83 return ' unknown '
84 casapath=os.environ["CASAPATH"].split()
85 versioninfo = sys.version_info
86 pyversion = '%s.%s'%(versioninfo[0],versioninfo[1])
87 if os.path.isdir(casapath[0]+'/'+casapath[1]+'/python/%s/asap'%(pyversion)):
88 # for casa developer environment (linux or darwin)
89 revinfo=casapath[0]+'/'+casapath[1]+'/python/%s/asap/svninfo.txt'%(pyversion)
90 else:
91 # for end-user environments
92 if casapath[1]=='darwin':
93 revinfo=casapath[0]+'/Resources/python/asap/svninfo.txt'
94 else:
95 revinfo=casapath[0]+'/lib/python%s/asap/svninfo.txt'%(pyversion)
96 if os.path.isfile(revinfo):
97 f = file(revinfo)
98 f.readline()
99 revsionno=f.readline()
100 f.close()
101 return revsionno.rstrip()
102 return ' unknown '
Note: See TracBrowser for help on using the repository browser.