def __init__(self, do_sanity_checks = True):
self.hostname = _gethostname()
self.user = _getuser()
self.is64bit = _architecture()[0] == '64bit'
self.is32bit = _architecture()[0] == '32bit'
if do_sanity_checks:
# sanity check
if self.is64bit + self.is32bit != 1:
raise ValueError(_architecture()[0])
self.has_good_gpu_card = True # Let's be optimistic.
def getDefaultElogInstance(self, url, **kwargs):
from pathlib import Path
home = str(Path.home())
if not ("user" in kwargs.keys()):
kwargs.update(dict(user=_getuser()))
if not ("password" in kwargs.keys()):
try:
with open(os.path.join(home, ".elog_psi"), "r") as f:
_pw = f.read().strip()
except:
print("Enter elog password for user: %s" % kwargs["user"])
_pw = _getpass()
kwargs.update(dict(password=_pw))
return _elog_ha.open(url, **kwargs), kwargs["user"]
def get_user():
"""Try to find the user who called sudo."""
try:
user = os.environ['USER']
except KeyError:
# possibly a systemd service. no sudo was used
return _getuser()
if user == 'root':
try:
return os.environ['SUDO_USER']
except KeyError:
# no sudo was used
pass
return user
def getDefaultElogInstance(url, **kwargs):
from pathlib import Path
home = str(Path.home())
if not ('user' in kwargs.keys()):
kwargs.update(dict(user=_getuser()))
if not ('password' in kwargs.keys()):
try:
with open(os.path.join(home, '.elog_psi'), 'r') as f:
_pw = f.read().strip()
except:
print('Enter elog password for user: %s' % kwargs['user'])
_pw = _getpass()
kwargs.update(dict(password=_pw))
return _elog_ha.open(url, **kwargs), kwargs['user']
def _update_env():
## Can't import modules, only concrete objects/functions.
## All vars (including imported things) need to have an underscore prefix if you want them ignored by invoke.
from os import environ as _environ
from os.path import join as _join, expanduser as _expanduser, exists as _exists
from getpass import getuser as _getuser
_release_auth_file = _expanduser(_join("~", "etc", "release.auth"))
if _exists(_release_auth_file):
_auth = open(_release_auth_file).read().strip().split(':')
_environ['WIKI_USER'] = _auth[0]
_environ['WIKI_PASS'] = _auth[1]
_environ['JIRA_USER'] = _auth[0]
_environ['JIRA_PASS'] = _auth[1]
_passwd_file = _expanduser(_join("~", "etc", "passwd"))
if _exists(_passwd_file):
_pass = open(_passwd_file).read().strip()
_environ['SMB_USER'] = _getuser()
_environ['SMB_PASS'] = _pass
def user(self):
return _getuser()
def __init__(self,screenshot_directory='',**kwargs):
self._screenshot_directory = screenshot_directory
if not ('user' in kwargs.keys()):
self.user=_getuser()
else:
self.user = kwargs['user']
def user(self):
return _getuser()
