添加 AbstrctDirectory 测试用例

Author: zhenwei-li

src/com/dvsnier/directory/abstract_directory.py

@@ -0,0 +1,67 @@
+# -*- coding:utf-8 -*-
+
+import logging
+import os
+
+from com.dvsnier.directory.idirectory import IDirectory
+
+
+class AbstractDirectory(IDirectory, object):
+    '''the Abstract Directory class'''
+    # True: the strict mode, otherwise the base mode
+    strategy_mode = False
+
+    def __init__(self, strategy_mode=False):
+        super(AbstractDirectory, self).__init__()
+        self.strategy_mode = strategy_mode
+        if self.strategy_mode:
+            logging.warning(
+                'At present, the strict review mode is adopted, and the illegal path is thrown directly. If the current directory is used by default to meet the current situation, please use the general review mode'
+            )
+        else:
+            logging.debug(
+                'Currently, the general review mode, workspace region space, current region space, executed region space and the same naming mode are used for the implementation of regional space'
+            )
+
+    def mk_dir(self, output_dir_name):
+        'the initialize global output dir that two modes are provided, one is `strict mode`, the other is `general mode`, and the default is `general mode` (i.e. current work execution environment)'
+        project_dir = None
+        if self.strategy_mode:
+            if self.get_work_region_space():
+                project_dir = self.get_work_region_space()
+            else:
+                raise ValueError(
+                    'In strict mode, illegal WORK_REGIOIN_SPACE value, please check WORK_REGIOIN_SPACE parameter.')
+        else:
+            project_dir = os.path.abspath('.')
+        output_dir = os.path.join(project_dir, output_dir_name)
+        if not os.path.exists(output_dir):
+            os.makedirs(output_dir)
+        return output_dir
+
+    def mk_children_dir(self, output_dir_name, sub_dir_name):
+        'the initialize children dir that default, the specified directory is created from the current execution environment'
+        if not os.path.exists(output_dir_name):
+            os.makedirs(output_dir_name)
+        children_dir = os.path.join(output_dir_name, sub_dir_name)
+        if not os.path.exists(children_dir):
+            os.makedirs(children_dir)
+        # logging.debug('the current children_dir is %s' % children_dir)
+        return children_dir
+
+    def mk_output_dir(self, output_dir_name, output_default_super_dir_name='out'):
+        'the initialize output dir that two modes are provided, one is `strict mode`, the other is `general mode`, and the default is `general mode` (i.e. current work execution environment)'
+        project_dir = None
+        if self.strategy_mode:
+            if self.get_work_region_space():
+                project_dir = self.get_work_region_space()
+            else:
+                raise ValueError(
+                    'In strict mode, illegal WORK_REGIOIN_SPACE value, please check WORK_REGIOIN_SPACE parameter.')
+        else:
+            project_dir = os.path.abspath('.')
+        out_dir = os.path.join(project_dir, output_default_super_dir_name)
+        output_dir = os.path.join(out_dir, output_dir_name)
+        if not os.path.exists(output_dir):
+            os.makedirs(output_dir)
+        return output_dir

src/com/dvsnier/directory/common_directory.py

@@ -1,11 +1,13 @@
 # -*- coding:utf-8 -*-
 
+import datetime
 import os
 import time
-import datetime
+
+from com.dvsnier.directory.abstract_directory import AbstractDirectory
 
 
-class CommonDirectory(object):
+class CommonDirectory(AbstractDirectory, object):
     '''the Common Directory class'''
     def __init__(self):
         super(CommonDirectory, self).__init__()
@@ -47,37 +49,3 @@ def generate_complex_or_fmt_file_name(self, output_dir_name, file_name, fmt='%Y%
         else:
             file_name = str("%s_%s" % (file_name, datetime.datetime.now().strftime(fmt)))
         return os.path.join(output_dir, file_name)
-
-    def mk_dir(self, output_dir_name):
-        'the initialize global output dir'
-        project_dir = os.path.abspath('.')
-        output_dir = os.path.join(project_dir, output_dir_name)
-        if not os.path.exists(output_dir):
-            os.makedirs(output_dir)
-        return output_dir
-
-    def mk_output_dir(self, output_dir_name):
-        'the initialize output dir'
-        # root_dir = os.path.dirname(os.path.abspath('.'))
-        # logging.debug('the current root_dir is %s' % root_dir)
-        project_dir = os.path.abspath('.')
-        # logging.debug('the current project_dir is %s' % project_dir)
-        # src_dir = os.path.join(project_dir, 'src')
-        # logging.debug('the current src_dir is %s' % src_dir)
-        out_dir = os.path.join(project_dir, 'out')
-        # logging.debug('the current out_dir is %s' % out_dir)
-        output_dir = os.path.join(out_dir, output_dir_name)
-        if not os.path.exists(output_dir):
-            os.makedirs(output_dir)
-        # logging.debug('the current output_dir is %s' % output_dir)
-        return output_dir
-
-    def mk_children_dir(self, output_dir_name, sub_dir_name):
-        'the initialize children dir'
-        if not os.path.exists(output_dir_name):
-            os.makedirs(output_dir_name)
-        children_dir = os.path.join(output_dir_name, sub_dir_name)
-        if not os.path.exists(children_dir):
-            os.makedirs(children_dir)
-        # logging.debug('the current children_dir is %s' % children_dir)
-        return children_dir

tests/com/dvsnier/directory/test_abstrct_directory.py

@@ -0,0 +1,56 @@
+# -*- coding:utf-8 -*-
+
+import os
+import unittest
+
+from com.dvsnier.directory.abstract_directory import AbstractDirectory
+
+
+class Test_AbstrctDirectory(unittest.TestCase):
+    ''' the test xxx '''
+    @classmethod
+    def setUpClass(cls):
+        print("...the set up...")
+        print('')
+        cls.directory = AbstractDirectory(strategy_mode=True)
+        # cls.directory = AbstractDirectory()
+
+    def setUp(self):
+        super(Test_AbstrctDirectory, self).setUp()
+
+    def test_0_mk_dir(self):
+        self.directory.set_work_region_space(os.path.join(os.getcwd(), 'out'))
+        self.directory.mk_dir('output/asc_0')
+
+    def test_1_mk_children_dir(self):
+        # the current executed environment directory
+        self.directory.mk_children_dir(output_dir_name='out', sub_dir_name='sub_1/asc_1')
+        # the settings work region space environment directory
+        self.directory.mk_children_dir(output_dir_name=self.directory.get_work_region_space(),
+                                       sub_dir_name='sub_2/asc_12')
+        # the settings executed region space environment directory
+        self.directory.mk_children_dir(output_dir_name=self.directory.get_executed_region_space(),
+                                       sub_dir_name='sub_3/asc_123')
+        # the settings current region space environment directory
+        self.directory.mk_children_dir(output_dir_name=self.directory.get_current_region_space(),
+                                       sub_dir_name='sub_4/asc_1234')
+
+    def test_2_mk_output_dir(self):
+        self.directory.mk_output_dir(output_dir_name=self.directory.get_work_region_space())
+        self.directory.mk_output_dir(output_dir_name=self.directory.get_work_region_space(),
+                                     output_default_super_dir_name='dist')
+
+    def tearDown(self):
+        super(Test_AbstrctDirectory, self).tearDown()
+
+    @classmethod
+    def tearDownClass(cls):
+        print('')
+        print("...the tear down...")
+
+
+if __name__ == '__main__':
+    ''' the unittest suite '''
+    suite = unittest.TestLoader().loadTestsFromTestCase(Test_AbstrctDirectory)
+    unittest.TextTestRunner(verbosity=2).run(suite)
+    # unittest.main()