getPath.py

Go to the documentation of this file.

 
 
 
 
import os
import copy
 
 
 
 
class Path:
    
    def __init__(self,mode):
        self._mode_mode=mode
        self._filePath_filePath=[]
        self._dirPath_dirPath=[]
        self._requirements_requirements=[] #同一个项目中可能有多个requirements,来自第三方库，如aepx
 
    
    @property
    def path(self):
        if self._mode_mode=='D':
            return copy.deepcopy(self._dirPath_dirPath)
        else:
            return copy.deepcopy(self._filePath_filePath)
 
    
    def clc(self):
        self._dirPath_dirPath.clear()
        self._filePath_filePath.clear()
 
    
    def getPath(self,rootDir):
        for root, dirs, files in os.walk(rootDir, followlinks=True):
            files=[f for f in files if f[0]!='.']  # 过滤掉以.开头的隐藏文件
            dirs=[d for d in dirs if d[0]!='.' and d!= '__pycache__' and d!='include']  # 过滤掉当前路径./和上一级路径../
            if self._mode_mode=='F':
                for f in files:
                    self._filePath_filePath.append(os.path.join(root,f))
                break
            
            elif self._mode_mode=='D':
                for dir in dirs:
                    self._dirPath_dirPath.append(os.path.join(root,dir))
                break
            
            elif self._mode_mode=='DF':
                for f in files:
                    if f.endswith('.py') or f.endswith('.pyi'): #在抽取库的时候需要.pyi,但在抽取项目代码时不需要.pyi文件
                        self._filePath_filePath.append(os.path.join(root, f))
                    if 'requirements.txt' in f:
                        self._requirements_requirements.append(os.path.join(root,f))
        
 
        #判断项目中是否有requirements.txt
        if len(self._requirements_requirements)>0:
            return self._requirements_requirements[0] #返回一个要求的版本号
        else:
            return None