码迷,mamicode.com
首页 > 其他好文 > 详细

Flask统计代码行数

时间:2019-01-03 20:41:49      阅读:285      评论:0      收藏:0      [点我收藏+]

标签:pfile   tree   adb   post   art   ==   strip()   strip   pac   

流程:

  1.获取前端的文件

  2.判断文件是否zip文件

  3.解压压缩包并保存

  4.遍历解压后的文件夹

  5.判断文件是否py文件,将绝对路径添加到列表

  6.循环列表,排除注释和空号,统计行数

from flask import Blueprint
from flask import render_template
from flask import request
from flask import current_app
import time
import os
import shutil

uploadBlue = Blueprint("uploadBlue",__name__)

@uploadBlue.route("/upload",methods=["GET","POST"])
def upload():
    error = ""
    if request.method == "POST":
        # 获取前端的上传的文件
        file = request.files.get("file")
        # 判断后缀是否zip的文件
        zip_file = file.filename.rsplit(".",1)
        if zip_file[-1] != "zip":
            return render_template("upload.html", error="上传的不是zip文件,请重新上传")

        # 解压保存
        t = int(time.time())
        upload_path = os.path.join(current_app.config.root_path,"files",str(t))
        try:
            shutil._unpack_zipfile(file,upload_path)        # 解压文件
        except Exception as e:
            print(e)
            shutil.rmtree(upload_path)          # 删除非空文件夹
            return render_template("upload.html", error="文件解压异常")

        # 遍历保存的文件夹得到所有.py文件
        file_list = []
        for (dirpath,dirname,filenames) in os.walk(upload_path):
            # 判断文件是否py结尾,如果是则将绝对路径添加到列表
            for file in filenames:
                if file.rsplit(".",1)[-1] == "py":
                    file_path = os.path.join(upload_path,file)
                    file_list.append(file_path)
        num = 0
        for pyfile in file_list:
            with open(pyfile,mode=rb) as f:
                for line in f:
                    # 井号开头或者空行则跳过
                    if line.replace(b" ",b"").strip().startswith(b"#") or not line.strip():
                        continue
                    num += 1
        return str(num)

    return render_template("upload.html",error=error)

 

Flask统计代码行数

标签:pfile   tree   adb   post   art   ==   strip()   strip   pac   

原文地址:https://www.cnblogs.com/st-st/p/10216621.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!