Add 0007 file

Drake-Z · Drake-Z · commit 0043b02553e2 · 2016-06-17T10:38:15.000+08:00
diff --git a/Drake-Z/0007/0007.py b/Drake-Z/0007/0007.py
@@ -0,0 +1,45 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+
+'第 0007 题：有个目录，里面是你自己写过的程序，统计一下你写过多少行代码。包括空行和注释，但是要分别列出来。'
+
+__author__ = 'Drake-Z'
+
+import os
+import re
+
+def count_num(a, b):
+    shuzi = [0, 0, 0]
+    path = os.path.join(a, b)
+    f = open(path, 'r', encoding='UTF-8').readlines()
+    for i in f:
+        if re.match(r'^#', i) == None:                          
+            pass
+        else:
+            shuzi[1] += 1                                                   #获得注释行数
+    if f[-1][-1:]=='\n':                                                     #最后一行为空行时
+        shuzi[2] = f.count('\n')+1                                     #获得空行行数
+        shuzi[0] = len(f)+1 - shuzi[2] - shuzi[1]                 #获得代码行数
+    else:
+        shuzi[2] = f.count('\n')
+        shuzi[0] = len(f) - shuzi[2] - shuzi[1]
+    return shuzi
+
+def file_analysis(c, d):
+    py = [x for x in os.listdir(c) if os.path.splitext(x)[1]==d]    #获得文件列表
+    print(py)
+    the_num = [0, 0, 0]
+    for i in py:
+        num = count_num(c, i)
+        the_num[0] += num[0]                #累计
+        the_num[1] += num[1]
+        the_num[2] += num[2]
+    print('''统计得目录中：
+             代码有 %s 行
+             注释 %s 行
+             空行 %s 行''' % (the_num[0], the_num[1], the_num[2]))
+
+if __name__ == '__main__':
+    file = '.'
+    ext = '.py'
+    file_analysis(file, ext)
diff --git a/Drake-Z/0007/test.py b/Drake-Z/0007/test.py
@@ -0,0 +1,41 @@
+# -*- coding: utf-8 -*-
+# mydict_test.py
+
+import unittest
+from MyMoudel.Moudel1 import Dict
+class TestDict(unittest.TestCase):
+
+    def test_init(self):
+        d = Dict(a=1, b='test')
+        self.assertEqual(d.a, 1)
+        self.assertEqual(d.b, 'test')
+        self.assertTrue(isinstance(d, dict))
+
+    def test_key(self):
+        d = Dict()
+        d['key'] = 'value'
+        self.assertEqual(d.key, 'value')
+
+    def test_attr(self):
+        d = Dict()
+        d.key = 'value'
+        self.assertTrue('key' in d)
+        self.assertEqual(d['key'], 'value')
+
+    def test_keyerror(self):
+        d = Dict()
+        with self.assertRaises(KeyError):
+            value = d['empty']
+
+    def test_attrerror(self):
+        d = Dict()
+        with self.assertRaises(AttributeError):
+            value = d.empty
+
+    def setUp(self):
+        print('setUp...')
+
+    def tearDown(self):
+        print('tearDown...')
+if __name__ == '__main__':
+    unittest.main()
