code_counter.py

#-*- coding:utf-8 -*-
'''
Created on 2013-9-2

@author: pkking
'''

import datetime
import os

class time(object):
    '''
    compute running time
    '''
    def __init__(self):
        '''
        Constructor
        '''
        
    def start(self):
        '''
        start count
        '''
        self.start_time = datetime.datetime.now()
        
    def end(self):
        '''
        end count
        '''
        self.end_time = datetime.datetime.now()
        
    def count_time(self):
        '''
        print the time it cost
        '''
        t = self.end_time - self.start_time
        print('running takes ' + str(t.days) + ' days ' + str(t.seconds)+' seconds and ' + str(t.microseconds)+' microsecond')       
        
class the_code(object):
    '''
    the code class
    '''
        
    def __init__(self):
        '''
        Constructor
        '''
        self.c_file_counter =0
        self.py_file_counter =0
        self.c_file_list = []
        self.py_file_list = [] 
        self.c_line_counter = 0
        self.py_line_counter = 0
        for root_dir,dirs,files in os.walk('.'):
                if(self.is_git_dir(root_dir)):
                    continue
                for this_file in files:
                    tail = os.path.splitext(this_file)
                    if(tail[1] == '.py'):
                        self.py_file_list.append(os.path.join(root_dir, this_file))
                        self.py_file_counter +=1
                    elif(tail[1] == '.c' or tail[1] == '.h'):
                        self.c_file_list.append(os.path.join(root_dir, this_file))
                        self.c_file_counter +=1

    def source_files_count(self, file_type):
        '''
        count code lines for c files
        '''
        if file_type == 'c' or file_type == 'C':
            for c_file in self.c_file_list:
                fobj = open(c_file,'r+')
                lines = fobj.readlines()
                for line in lines:
                    if(';' in line or '#' in line):
                        self.c_line_counter +=1
                fobj.close()
        elif file_type == 'py' or file_type == 'PY' or file_type == 'pY' or file_type == 'Py':
            for py_file in self.py_file_list:
                fobj = open(py_file,'r+')
                lines = fobj.readlines()
                for line in lines:
                    if( '\r\n' in line or '\n' in line or '\r' in line):
                        self.py_line_counter +=1
                fobj.close()
        else:
            print('no this kind source file!')
            exit()
            
    def is_git_dir(self,source_dir):
        '''
        if the dir is a git repo
        '''
        if '.git' == source_dir:
            return True
        else:
            return False
    
    def print_line(self):
        '''
        print the line num
        '''
        if(self.c_file_counter):
            print'there are ' + str(self.c_file_counter) + ' C source files'
            print'the c and h source files lines are :' + str(self.c_line_counter)
        if(self.py_file_counter):
            print'there are ' + str(self.py_file_counter) + ' Python source files'
            print'the Python source files lines are:' + str(self.py_line_counter)

t = time()
t.start()
my_code = the_code()
my_code.source_files_count('c')
my_code.source_files_count('py')
t.end()
my_code.print_line()
t.count_time()
raw_input("Press Enter to continue...")