102 lines
3.1 KiB
Python
102 lines
3.1 KiB
Python
|
# *****************************************************************************
|
||
|
|
#
|
||
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
||
|
|
# you may not use this file except in compliance with the License.
|
||
|
|
# You may obtain a copy of the License at
|
||
|
|
#
|
||
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
||
|
|
#
|
||
|
|
# Unless required by applicable law or agreed to in writing, software
|
||
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
||
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||
|
|
# See the License for the specific language governing permissions and
|
||
|
|
# limitations under the License.
|
||
|
|
#
|
||
|
|
# See NOTICE file for details.
|
||
|
|
#
|
||
|
|
# *****************************************************************************
|
||
|
|
import os as _os
|
||
|
|
import typing
|
||
|
|
|
||
|
|
import _jpype
|
||
|
|
|
||
|
|
__all__ = ['addClassPath', 'getClassPath']
|
||
|
|
|
||
|
|
_CLASSPATHS = []
|
||
|
|
_SEP = _os.path.pathsep
|
||
|
|
|
||
|
|
|
||
|
|
def addClassPath(path1: typing.Union[str, _os.PathLike]) -> None:
|
||
|
|
""" Add a path to the Java class path
|
||
|
|
|
||
|
|
Classpath items can be a java, a directory, or a
|
||
|
|
glob pattern. Relative paths are relative to the
|
||
|
|
caller location.
|
||
|
|
|
||
|
|
Arguments:
|
||
|
|
path(str):
|
||
|
|
|
||
|
|
"""
|
||
|
|
# We are deferring these imports until here as we only need them
|
||
|
|
# if this function is used.
|
||
|
|
from pathlib import Path
|
||
|
|
import inspect
|
||
|
|
global _CLASSPATHS
|
||
|
|
|
||
|
|
# Convert to an absolute path. Note that
|
||
|
|
# relative paths will be resolve based on the location
|
||
|
|
# of the caller rather than the JPype directory.
|
||
|
|
path1 = Path(path1)
|
||
|
|
if not path1.is_absolute():
|
||
|
|
path2 = Path(inspect.stack(1)[1].filename).parent.resolve()
|
||
|
|
path1 = path2.joinpath(path1)
|
||
|
|
|
||
|
|
# If the JVM is already started then we will have to load the paths
|
||
|
|
# immediately into the DynamicClassLoader
|
||
|
|
if _jpype.isStarted():
|
||
|
|
Paths = _jpype.JClass('java.nio.file.Paths')
|
||
|
|
JContext = _jpype.JClass('org.jpype.JPypeContext')
|
||
|
|
classLoader = JContext.getInstance().getClassLoader()
|
||
|
|
if path1.name == "*":
|
||
|
|
paths = list(path1.parent.glob("*.jar"))
|
||
|
|
if len(paths) == 0:
|
||
|
|
return
|
||
|
|
for path in paths:
|
||
|
|
classLoader.addFile(Paths.get(str(path)))
|
||
|
|
else:
|
||
|
|
classLoader.addFile(Paths.get(str(path1)))
|
||
|
|
_CLASSPATHS.append(path1)
|
||
|
|
|
||
|
|
|
||
|
|
def getClassPath(env: bool = True) -> str:
|
||
|
|
""" Get the full Java class path.
|
||
|
|
|
||
|
|
Includes user added paths and the environment CLASSPATH.
|
||
|
|
|
||
|
|
Arguments:
|
||
|
|
env(Optional, bool): If true then environment is included.
|
||
|
|
(default True)
|
||
|
|
"""
|
||
|
|
from pathlib import Path
|
||
|
|
global _CLASSPATHS
|
||
|
|
global _SEP
|
||
|
|
|
||
|
|
# Merge the environment path
|
||
|
|
classPath = list(_CLASSPATHS)
|
||
|
|
envPath = _os.environ.get("CLASSPATH")
|
||
|
|
if env and envPath:
|
||
|
|
classPath.extend([Path(i) for i in envPath.split(_SEP)])
|
||
|
|
|
||
|
|
out = []
|
||
|
|
for path in classPath:
|
||
|
|
if path == '':
|
||
|
|
continue
|
||
|
|
if path.name == "*":
|
||
|
|
paths = list(path.parent.glob("*.jar"))
|
||
|
|
if len(paths) == 0:
|
||
|
|
continue
|
||
|
|
out.extend(paths)
|
||
|
|
else:
|
||
|
|
out.append(path)
|
||
|
|
return _SEP.join([str(i) for i in out])
|