添加新版本 CommonDirectory

Author: zhenwei-li

src/com/dvsnier/directory/common_directory.py

@@ -0,0 +1,83 @@
+# -*- coding:utf-8 -*-
+
+import os
+import time
+import datetime
+
+
+class CommonDirectory(object):
+    '''the Common Directory class'''
+    def __init__(self):
+        super(CommonDirectory, self).__init__()
+
+    def generate_complex_file_name(self, output_dir_name, file_name):
+        'the generate complex file name'
+        output_dir = self.mk_output_dir(output_dir_name)
+        name = str("%s_%s.log" % (file_name, int(time.time())))
+        return os.path.join(output_dir, name)
+
+    def generate_fmt_file_name(self, output_dir_name, file_name, fmt='%Y%m%d_%H%M%S'):
+        'the generate out with fmt file name'
+        output_dir = self.mk_output_dir(output_dir_name)
+        name = str("%s_%s.txt" % (file_name, datetime.datetime.now().strftime(fmt)))
+        return os.path.join(output_dir, name)
+
+    def generate_file_name(self, output_dir_name, file_name):
+        'the generate out file name'
+        output_dir = self.mk_output_dir(output_dir_name)
+        return os.path.join(output_dir, file_name)
+
+    def generate_file_name_only(self, output_dir_name, file_name):
+        'the generate file name only'
+        output_dir = self.mk_dir(output_dir_name)
+        return os.path.join(output_dir, file_name)
+
+    def generate_complex_or_fmt_file_name(self, output_dir_name, file_name, fmt='%Y%m%d_%H%M%S'):
+        'the generate out complex or fmt file name'
+        output_dir = self.mk_output_dir(output_dir_name)
+        if file_name is None or len(file_name.strip()) == 0:
+            raise ValueError('the file name is invaild.')
+        file_name = file_name.strip()
+        if '.' in file_name:
+            rdot_index = file_name.rfind('.')
+            if rdot_index > 0:
+                file_name = str(
+                    "%s_%s%s" %
+                    (file_name[0:rdot_index], datetime.datetime.now().strftime(fmt), file_name[rdot_index:]))
+        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_common_directory.py

@@ -0,0 +1,103 @@
+# -*- coding:utf-8 -*-
+
+import unittest
+
+from com.dvsnier.directory.common_directory import CommonDirectory
+
+
+class Test_Common_Directory(unittest.TestCase):
+    ''' the test common dir '''
+    @classmethod
+    def setUpClass(cls):
+        print("...the set up...")
+        print('')
+        cls.directory = CommonDirectory()
+
+    def setUp(self):
+        super(Test_Common_Directory, self).setUp()
+
+    def test_generate_complex_file_name(self):
+        'the test generate complex file name'
+        dir_name = 'https'
+        file_name = 'test_name'
+        output = self.directory.generate_complex_file_name(dir_name, file_name)
+        print("\nthe test generate_complex_file_name(%s) is succeed." % output)
+        self.assertIsNotNone(output, 'test_generate_complex_file_name is error.')
+
+    def test_generate_fmt_file_name(self):
+        'the test generate fmt file name'
+        dir_name = 'https'
+        file_name = 'test_name'
+        # output = generate_fmt_file_name(dir_name, file_name)
+        output = self.directory.generate_fmt_file_name(dir_name, file_name, fmt='%Y%m%d_%H%M%S.%f')
+        print("\nthe test generate_fmt_file_name(%s) is succeed." % output)
+        self.assertIsNotNone(output, 'generate_fmt_file_name is error.')
+
+    def test_generate_file_name(self):
+        'the test generate file name'
+        dir_name = 'https'
+        file_name = 'test_name'
+        output = self.directory.generate_file_name(dir_name, file_name)
+        print("\nthe test test_generate_file_name(%s) is succeed." % output)
+        self.assertIsNotNone(output, 'test_generate_file_name is error.')
+
+    def test_generate_file_name_only(self):
+        'the test generate file name only'
+        dir_name = 'out/https_2'
+        file_name = 'test_name'
+        output = self.directory.generate_file_name_only(dir_name, file_name)
+        print("\nthe test test_generate_file_name_only(%s) is succeed." % output)
+        self.assertIsNotNone(output, 'test_generate_file_name_only is error.')
+
+    def test_generate_complex_or_fmt_file_name(self):
+        'the test generate complex or fmt file name'
+        dir_name = 'https'
+        # file_name = ''
+        # file_name = ' '
+        # file_name = 'test_name'
+        # file_name = ' test_name'
+        # file_name = 'test_name '
+        # file_name = ' test_name '
+        file_name = 'test_name.txt'
+        # file_name = 'test_name.txt.log'
+        # file_name = '.test_name.txt.log'
+        # file_name = '.test_name.txt.log.'
+        # file_name = 'test.name.txt.log'
+        output = self.directory.generate_complex_or_fmt_file_name(dir_name, file_name)
+        print("\nthe test test_generate_complex_or_fmt_file_name(%s) is succeed." % output)
+        self.assertIsNotNone(output, 'test_generate_complex_or_fmt_file_name is error.')
+
+    def test_mk_dir(self):
+        'the test mk dir'
+        dir_name = 'config'
+        output = self.directory.mk_dir(dir_name)
+        print("\nthe test test_mk_dir(%s) is succeed." % output)
+        self.assertIsNotNone(output, 'test_mk_dir is error.')
+
+    def test_mk_output_dir(self):
+        'the test mk output dir'
+        dir_name = 'https'
+        output = self.directory.mk_output_dir(dir_name)
+        print("\nthe test mk_output_dir(%s) is succeed." % output)
+        self.assertIsNotNone(output, 'test_mk_output_dir is error.')
+
+    def test_mk_children_dir(self):
+        'the test mk childer dir'
+        sub_name = 'children'
+        output = self.directory.mk_children_dir('out/http', sub_dir_name=sub_name)
+        print("\nthe test mk_children_dir(%s) is succeed." % output)
+        self.assertIsNotNone(output, 'test_mk_children_dir is error.')
+
+    def tearDown(self):
+        super(Test_Common_Directory, self).tearDown()
+
+    @classmethod
+    def tearDownClass(cls):
+        print('')
+        print("...the tear down...")
+
+
+if __name__ == '__main__':
+    suite = unittest.TestLoader().loadTestsFromTestCase(Test_Common_Directory)
+    unittest.TextTestRunner(verbosity=2).run(suite)
+    # unittest.main()