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
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
|
# -*- coding: UTF-8 -*-
# Copyright 2004-2005 Gentoo Foundation
# Distributed under the terms of the GNU General Public License v2
# $Header: $
from Package import Package
from VM import VM
from Errors import *
from os.path import basename, dirname
from glob import glob
import os, re
class EnvironmentManager:
virtual_machines = None
packages = None
active = None
vms_path = '/etc/env.d/java/'
pkg_path = '/usr/share/*/package.env'
def __init__(self):
pass
def load_vms(self):
self.virtual_machines = {}
if os.path.isdir(self.vms_path):
count = 1
for file in os.listdir(self.vms_path):
if file.startswith("20"):
conf = os.path.join(self.vms_path,file)
vm = None
try:
vm = VM(conf)
except InvalidConfigError:
pass
except PermissionError:
pass
self.virtual_machines[count] = vm
count += 1
def load_packages(self):
self.packages = []
for package in iter(glob(self.pkg_path)):
self.packages.append(Package(package, basename(dirname(package))))
def load_active_vm(self):
for link in self.vm_links():
if os.path.islink(link):
vm_name = os.readlink(link)
vm = self.get_vm(vm_name)
self.active = vm
return vm
raise InvalidVMError
def set_active_vm(self, vm):
self.active = vm
def get_active_vm(self):
if self.active is None:
self.load_active_vm()
return self.active
def get_virtual_machines(self):
if self.virtual_machines is None:
self.load_vms()
return self.virtual_machines
def find_vm(self, name):
found = []
for id, vm in self.get_virtual_machines().iteritems():
if vm.name().startswith(name):
found.append(vm)
return found
def get_packages(self):
if self.packages is None:
self.load_packages()
return self.packages
def query_packages(self, packages, query):
results = []
for package in iter(self.get_packages()):
if package.name() in packages:
value = package.query(query)
if value:
results.append(value)
packages.remove(package.name())
return results
def get_vm(self, machine):
vm_list = self.get_virtual_machines()
selected = None
for count in iter(vm_list):
vm = vm_list[count]
if str(machine).isdigit():
if int(machine) is count:
return vm
else:
# Check if the vm is specified via env file
if machine == vm.filename():
return vm
# Check if the vm is specified by name
elif machine == vm.name():
return vm
# Check if the vm is specified via JAVA_HOME
elif machine == vm.query('JAVA_HOME'):
return vm
# Check if vm is specified by partial name
elif vm.name().startswith(machine):
selected = vm
if selected:
return selected
else:
return None
def set_user_vm(self, vm):
self.set_vm(vm, self.user_vm_link())
def set_system_vm(self, vm):
self.set_vm(vm, self.system_vm_link())
def set_vm(self, vm, target):
if os.path.islink(target):
os.remove(target)
os.symlink(vm.name(),target)
def vm_links(self):
return [ self.user_vm_link(), self.system_vm_link() ]
def user_vm_link(self):
return os.path.join(os.environ.get('HOME'), '.gentoo/user-vm')
def system_vm_link(self):
return '/usr/share/java-config/vms/system-vm'
def clean_classpath(self, env_file):
if os.path.isfile(env_file):
try:
os.remove(env_file)
except IOError:
raise PermissionError
def set_classpath(self, env_file, pkgs):
classpath = self.query_packages(pkgs, "CLASSPATH")
classpath = re.sub(':+', ':', ':'.join(classpath)).strip(':')
self.clean_classpath(env_file)
self.write_classpath(env_file, classpath)
def append_classpath(self, env_file, pkgs):
classpath = self.query_packages(pkgs, "CLASSPATH")
classpath = re.sub(':+', ':', ':'.join(classpath)).strip(':')
oldClasspath = ''
if os.path.isfile(env_file):
try:
stream = open(env_file, 'r')
except IOError:
raise PermissionError
read = stream.readline()
while read:
if read.strip().startswith('CLASSPATH'):
stream.close()
oldClasspath = read.split('=', 1)[-1].strip()
break
else:
read = stream.readline()
stream.close()
classpath = oldClasspath + ':' + classpath
self.clean_classpath(env_file)
self.write_classpath(env_file, classpath)
def write_classpath(self, env_file, classpath):
try:
stream = open(env_file, 'w')
except IOError:
raise PermissionError
stream.write("CLASSPATH=%s\n" % (classpath))
stream.close()
# vim:set expandtab tabstop=4 shiftwidth=4 softtabstop=4 nowrap:
|
GitWeb