#!/usr/bin/env python
# -*- coding:utf-8 -*-
#導(dǎo)入正則模塊
import re
#導(dǎo)入os模塊，本例用它來做文件寫入操作
import os
#導(dǎo)入日志模塊
import logging
import logging.handlers
import xlwt
import xlrd
import MySQLdb
import pymysql

db = pymysql.connect("localhost","root","123456","my_db_hanchuan01" )

# 遍歷文件+采集手機(jī)號(hào)的方法
def count(dirname='D:/tmp/臨時(shí)數(shù)據(jù)源/sheet轉(zhuǎn)換/txtfiles'):
  #-----------------------正則--------------------------------  #手機(jī)號(hào)正則
  pattern_mob = re.compile('1[345789]\d{9}') 
  # pattern_mob_plus = re.sub('1[345789]\d{9}','1[345789]\d{9}+123')
  #---------------------統(tǒng)計(jì)文件------------------------------   # 記錄文件總數(shù)的變量
  count=0
  num=0 # 記錄含手機(jī)號(hào)文件數(shù)的變量
  for root, dirs, files in os.walk(dirname):
    # 得到每個(gè)文件夾下文件數(shù)量
    for filename in files:
      # 將文件名與路徑拼接成全路徑
      f = open(dirname+'/'+filename,'r+', encoding='utf-8')      
      var1 = (filename) # 將文件名寫入數(shù)據(jù)庫
      result = pattern_mob.findall(f.read())  # result = pattern_mob_plus.findall(f.read()) # result = pattern_mob.findall # result = result+"123"
      for phonenames in result:      
       var2 = (phonenames)
       print(var1)  
       print(var2)  
       cursor = db.cursor()  # 使用 cursor() 方法創(chuàng)建一個(gè)游標(biāo)對(duì)象 cursor
       cursor.execute("INSERT INTO 00_HEFEI_PHONES VALUES('%s', %s)" % (var1,var2)) # 執(zhí)行sql語句
       db.commit()        
# 主函數(shù)
   
if __name__ == '__main__':
#   # 調(diào)用count()h函數(shù)
  count()
  db.close()