__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 行''' % (the_num[0], the_num[1], the_num[2]))

if __name__ == '__main__':

file = '.'

ext = '.py'

file_analysis(file, ext)

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()