Format adjustment

tests/get_catalog.py

@@ -21,7 +21,7 @@ sys.path.append("pytest")
 import importlib
 
 class CatalogGen:
-    def __init__(self,CaseDirList,CatalogName,DirDepth):
+    def __init__(self, CaseDirList, CatalogName, DirDepth):
         self.CaseDirList = CaseDirList
         self.CatalogName = CatalogName
         self.blacklist = []
@@ -32,17 +32,18 @@ class CatalogGen:
             self.GetCatalog(i)
             self.CollectLog(i)
         print('Catalog Generation done')
-    def CollectLog(self,CaseDir):
+
+    def CollectLog(self, CaseDir):
         DirorFiles = os.listdir(CaseDir)
         for loop in range(self.DirDepth):
             for i in DirorFiles:
-                fileName = os.path.join(CaseDir,i)
+                fileName = os.path.join(CaseDir, i)
                 if os.path.isdir(fileName):
                     self.CollectLog(fileName)
                 else:
                     if i == self.CatalogName and fileName not in self.blacklist:
                         self.blacklist.append(fileName)
-                        with open(fileName,"r") as f :
+                        with open(fileName, "r") as f :
                             Catalog = f.read()
                         title = CaseDir.split("/")[-1]
                         TitleLevel = CaseDir.count("/")
@@ -50,20 +51,18 @@ class CatalogGen:
                             f.write("#" * TitleLevel + ' %s\n' % title)
                             f.write(Catalog)
 
-
-
-    def GetCatalog(self,CaseDir):
-        for root,dirs,files in os.walk(CaseDir):
+    def GetCatalog(self, CaseDir):
+        for root, dirs, files in os.walk(CaseDir):
             for file in files:
                 if file.endswith(".py"):
-                    fileName = os.path.join(root,file)
+                    fileName = os.path.join(root, file)
                     moduleName = fileName.replace(".py", "").replace("/", ".")
                     uModule = importlib.import_module(moduleName)
                     title = file.split('.')[0]
                     TitleLevel = root.count('/') + 1
                     try:
                         ucase = uModule.TDTestCase()
-                        with open(root + '/' + self.CatalogName,'a') as f:
+                        with open(root + '/' + self.CatalogName, 'a') as f:
                             f.write('#'*TitleLevel + ' %s\n' % title)
                             for i in ucase.caseDescription.__doc__.split('\n'):
                                 if i.lstrip() == '':continue
@@ -73,7 +72,7 @@ class CatalogGen:
 
     def CleanCatalog(self):
         for i in self.CaseDirList:
-            for root,dirs,files in os.walk(i):
+            for root, dirs, files in os.walk(i):
                 for file in files:
                     if file == self.CatalogName:
                         os.remove(root + '/' + self.CatalogName)
@@ -90,8 +89,8 @@ if __name__ == "__main__":
         if key in ['-h', '--help']:
             print(
                 'A collection of test cases catalog written using Python')
-            print("-d root dir of test case files written by Python,default: system-test,develop-test")
-            print('-c catalog file name,default: catalog.md')
+            print("-d root dir of test case files written by Python, default: system-test,develop-test")
+            print('-c catalog file name, default: catalog.md')
             print('-v dir depth of test cases.default: 5')
             print('-n <True:False> generate')
             print('-r <True:False> delete')
@@ -118,20 +117,16 @@ if __name__ == "__main__":
             elif (value.upper() == "FALSE"):
                 delete = False
 
-        
-
-    # CatalogName = 'catalog.md'
-    # CaseDirList = ['system-test','develop-test']
-    print(CaseDirList,CatalogName)
+    print(CaseDirList, CatalogName)
     if CaseDirList == [] :
-        CaseDirList = ['system-test','develop-test']
+        CaseDirList = ['system-test', 'develop-test']
     if CatalogName == '' :
         CatalogName = 'catalog.md'
     if DirDepth == 0:
         DirDepth = 5
     print('opt:\n\tcatalogname: %s\n\tcasedirlist: %s\n\tdepth: %d\n\tgenerate: %s\n\tdelete: %s' 
-            % (CatalogName,','.join(CaseDirList),DirDepth,generate,delete))
-    f = CatalogGen(CaseDirList,CatalogName,DirDepth)
+            % (CatalogName, ','.join(CaseDirList), DirDepth, generate, delete))
+    f = CatalogGen(CaseDirList, CatalogName, DirDepth)
     if delete:
         f.CleanCatalog()
     if generate: