Преглед на файлове

HUE-3716 [core] Add gen-py paths to hue.pth

Jenny Kim преди 9 години
родител
ревизия
247788efd9
променени са 1 файла, в които са добавени 12 реда и са изтрити 0 реда
  1. 12 0
      tools/app_reg/pth.py

+ 12 - 0
tools/app_reg/pth.py

@@ -68,6 +68,12 @@ class PthFile(object):
       self._entries.add(module_path)
       self._entries.add(module_path)
       LOG.debug('Add to %s: %s' % (self._path, module_path))
       LOG.debug('Add to %s: %s' % (self._path, module_path))
 
 
+      # Add gen-py path if found
+      gen_py_path = os.path.join(app.abs_path, 'gen-py')
+      if os.path.exists(gen_py_path):
+        self._entries.add(gen_py_path)
+        LOG.debug('Add to %s: %s' % (self._path, gen_py_path))
+
       # Eggs could be in ext-py/<pkg>/dist/*.egg
       # Eggs could be in ext-py/<pkg>/dist/*.egg
       for py in app.find_ext_pys():
       for py in app.find_ext_pys():
         ext_eggs = glob.glob(os.path.join(py, 'dist', '*.egg'))
         ext_eggs = glob.glob(os.path.join(py, 'dist', '*.egg'))
@@ -86,6 +92,12 @@ class PthFile(object):
       self._entries.add(module_path)
       self._entries.add(module_path)
       LOG.debug('Add to %s: %s' % (self._path, module_path))
       LOG.debug('Add to %s: %s' % (self._path, module_path))
 
 
+      # Add gen-py path if found
+      gen_py_path = self._relpath(os.path.join(app.abs_path, 'gen-py'))
+      if os.path.exists(os.path.join(app.abs_path, 'gen-py')):
+        self._entries.add(gen_py_path)
+        LOG.debug('Add to %s: %s' % (self._path, gen_py_path))
+
       # Eggs could be in ext-py/<pkg>/dist/*.egg
       # Eggs could be in ext-py/<pkg>/dist/*.egg
       for py in app.find_ext_pys():
       for py in app.find_ext_pys():
         ext_eggs = [self._relpath(egg) for egg in glob.glob(os.path.join(py, 'dist', '*.egg'))]
         ext_eggs = [self._relpath(egg) for egg in glob.glob(os.path.join(py, 'dist', '*.egg'))]