-
Notifications
You must be signed in to change notification settings - Fork 4.6k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
[Pythonclient] Add missing setup_path.py in environment folder
- Loading branch information
Showing
1 changed file
with
52 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,52 @@ | ||
# Import this module to automatically setup path to local airsim module | ||
# This module first tries to see if airsim module is installed via pip | ||
# If it does then we don't do anything else | ||
# Else we look up grand-parent folder to see if it has airsim folder | ||
# and if it does then we add that in sys.path | ||
|
||
import os,sys,inspect,logging | ||
|
||
#this class simply tries to see if airsim | ||
class SetupPath: | ||
@staticmethod | ||
def getDirLevels(path): | ||
path_norm = os.path.normpath(path) | ||
return len(path_norm.split(os.sep)) | ||
|
||
@staticmethod | ||
def getCurrentPath(): | ||
cur_filepath = os.path.abspath(inspect.getfile(inspect.currentframe())) | ||
return os.path.dirname(cur_filepath) | ||
|
||
@staticmethod | ||
def getGrandParentDir(): | ||
cur_path = SetupPath.getCurrentPath() | ||
if SetupPath.getDirLevels(cur_path) >= 2: | ||
return os.path.dirname(os.path.dirname(cur_path)) | ||
return '' | ||
|
||
@staticmethod | ||
def getParentDir(): | ||
cur_path = SetupPath.getCurrentPath() | ||
if SetupPath.getDirLevels(cur_path) >= 1: | ||
return os.path.dirname(cur_path) | ||
return '' | ||
|
||
@staticmethod | ||
def addAirSimModulePath(): | ||
# if airsim module is installed then don't do anything else | ||
#import pkgutil | ||
#airsim_loader = pkgutil.find_loader('airsim') | ||
#if airsim_loader is not None: | ||
# return | ||
|
||
parent = SetupPath.getParentDir() | ||
if parent != '': | ||
airsim_path = os.path.join(parent, 'airsim') | ||
client_path = os.path.join(airsim_path, 'client.py') | ||
if os.path.exists(client_path): | ||
sys.path.insert(0, parent) | ||
else: | ||
logging.warning("airsim module not found in parent folder. Using installed package (pip install airsim).") | ||
|
||
SetupPath.addAirSimModulePath() |