1. 程式人生 > 程式設計 >基於python實現對檔案進行切分行

基於python實現對檔案進行切分行

針對配置檔案進行切分,重組,每隔30行為一段,進行重新生成功能。

程式碼如下

#!/usr/local/python/bin/python
# coding=utf-8
import sys
import re
import os
f = open('config.conf','r')
#判斷檔案條數
def file_num(filename):
  num_col = 0
  with open(filename,'rb') as Fnum:
    while(Fnum.readline() !=''):
     num_col = num_col + 1
  return num_col
  

file_num = file_num('config.conf')
print '檔案總條數:' + str(file_num)

i = 0 #設定計數器
while i < file_num :       #表示檔案行數
  with open('/tmp/newfile/newfile'+str(i),'w') as f1:
    for j in range(0,30) : #這裡設定每個子檔案的大小
      if i <= file_num : #這裡判斷是否已結束,否則最後可能報錯
        f1.writelines(f.readline())
        i = i+1
      else:
        break

filepath1='/tmp/newfile/'
def eachFile(filepath):
  num = 1
  pathDir = sorted(os.listdir(filepath))
  for allDir in pathDir:
    child = os.path.join('%s%s' % (filepath,allDir))
    file_name = child.decode('gbk')
    print file_name
    with open('/tmp/check.py','a+') as f2:
     fopen = open(file_name,'r+')
     f2.writelines("# config.py sql" + str(num) + " begin\n")
     count=0
     for line in fopen.readlines(): 
       if count == 0:
        f2.writelines(line)
        count= count+1
       else:
        f2.writelines("UNION ALL " + line)
     f2.writelines("# config.py sql" + str(num) + " end\n")
     f2.writelines("\n")
     num = num + 1

eachFile(filepath1)

指令碼說明:

目錄路徑:/tmp

配置檔案:config.conf

準備目錄:/tmp/newfile/

最終新的配置檔案為:check.py

以上就是本文的全部內容,希望對大家的學習有所幫助,也希望大家多多支援我們。