FilesExpand file tree

set_environment_support

/

common.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

103 lines (76 loc) · 3.1 KB

set_environment_support

/

common.py

Copy path

Top

File metadata and controls

• Code
• Blame

103 lines (76 loc) · 3.1 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

62

63

64

65

66

67

68

69

70

71

72

73

74

75

76

77

78

79

80

81

82

83

84

85

86

87

88

89

90

91

92

93

94

95

96

97

98

99

100

101

102

103

class EnvironmentMixin(object):

"""Mixin class for managing session and server level environment

variables in tmux.

"""

_add_option = None

def __init__(self, add_option=None):

self._add_option = add_option

def set_environment(self, name, value):

"""Set environment ``$ tmux set-environment <name> <value>``.

:param name: the environment variable name. such as 'PATH'.

:type option: string

:param value: environment value.

:type value: string

"""

args = ['set-environment']

if self._add_option:

args += [self._add_option]

args += [name, value]

proc = self.cmd(*args)

if proc.stderr:

if isinstance(proc.stderr, list) and len(proc.stderr) == int(1):

proc.stderr = proc.stderr[0]

raise ValueError('tmux set-environment stderr: %s' % proc.stderr)

def unset_environment(self, name):

"""Unset environment variable ``$ tmux set-environment -u <name>``.

:param name: the environment variable name. such as 'PATH'.

:type option: string

"""

args = ['set-environment']

if self._add_option:

args += [self._add_option]

args += ['-u', name]

proc = self.cmd(*args)

if proc.stderr:

if isinstance(proc.stderr, list) and len(proc.stderr) == int(1):

proc.stderr = proc.stderr[0]

raise ValueError('tmux set-environment stderr: %s' % proc.stderr)

def remove_environment(self, name):

"""Remove environment variable ``$ tmux set-environment -r <name>``.

:param name: the environment variable name. such as 'PATH'.

:type option: string

"""

args = ['set-environment']

if self._add_option:

args += [self._add_option]

args += ['-r', name]

proc = self.cmd(*args)

if proc.stderr:

if isinstance(proc.stderr, list) and len(proc.stderr) == int(1):

proc.stderr = proc.stderr[0]

raise ValueError('tmux set-environment stderr: %s' % proc.stderr)

def show_environment(self, name=None):

"""Show environment ``$tmux show-environment -t [session] <name>``.

Return dict of environment variables for the session or the value of a

specific variable if the name is specified.

:param name: the environment variable name. such as 'PATH'.

:type option: string

"""

tmux_args = ['show-environment']

if self._add_option:

tmux_args += [self._add_option]

if name:

tmux_args += [name]

vars = self.cmd(*tmux_args).stdout

vars = [tuple(item.split('=', 1)) for item in vars]

vars_dict = {}

for t in vars:

if len(t) == 2:

vars_dict[t[0]] = t[1]

elif len(t) == 1:

vars_dict[t[0]] = True

else:

raise ValueError('unexpected variable %s', t)

if name:

return vars_dict.get(name)

return vars_dict