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']