用python将项目中的所有代码(或txt)合并在一个文件中

设计模式开卷考试给的例子代码都是一个类一个java,实在太恶心了,所以写了一个python脚本.

import os
fileansdir=r'C:\Users\tonyson_in_the_rain\Desktop\笔记\JAVA代码\设计模式'

主要用的os库

API总结:

files=listdir(path) 显示当前目录的所有文件名和目录名
endswith() 选择扩展名
windows的路径只需要前面加一个r即可在python程序中使用
把合并的文件在当前目录保存,也可以改成某一目录保存,但是要注意名字

combine函数是一个子过程,可以把dirpath目录下的所有文件合并为一个文件

def combine(dirPath,filename):
    files=os.listdir(dirPath)
    res=""
    i = 0
    for file in files:
        if file.endswith(".java"):
            i+=1
            title=file            with open(dirPath+'\\'+file,"r")as file:
                content=file.read()
                file.close()
            res+='\n\n'            res+='#### '+title
            res+='\n\n'
            res+='```\n'
            res+=content
            res+='\n```\n'            with open(dirPath+'\\'+filename+'.txt','w')as outFile:
                outFile.write(res)
                outFile.close()

然后遍历各个文件夹分别跑一遍子过程即可

import osdef EnumPathFiles(path, callback):
    ii=0
    if not os.path.isdir(path):
        print('Error:"',path,'" is not a directory or does not exist.')
        return
    list_dirs = os.walk(path)    for root, dirs, files in list_dirs:        for d in dirs:
            EnumPathFiles(os.path.join(root, d), callback)
            dirc=path+'\\'+d
            print(dirc)
            if(dirc.endswith('memento')):
                continue
            ii+=1
            name=d            combine(dirc,name)def callback1(path, filename):
    print(path+'\\'+filename)EnumPathFiles(r'C:\Users\tonyson_in_the_rain\Desktop\大三上学习\设计模式\设计模式2018(张欣佳)', callback1)

这里会报一个错误,因为子目录对不上号的原因,不过在这之前有用的目录我们已经跑过了,所以不用管了.

合成的md文件如下:不过是按照类名字典序排的,需要手动调整顺序

Apple.java

public class Apple implements Product
{
  public void accept(Visitor visitor)
  {
      visitor.visit(this);
  }
}

Book.java

public class Book implements Product
{
  public void accept(Visitor visitor)
  {
      visitor.visit(this);
  }
}

BuyBasket.java

import java.util.*;public class BuyBasket
{
private ArrayList list=new ArrayList();public void accept(Visitor visitor)
{
Iterator i=list.iterator();while(i.hasNext())
{
((Product)i.next()).accept(visitor);
}
}public void addProduct(Product product)
{
list.add(product);
}public void removeProduct(Product product)
{
list.remove(product);
}
}

Client.java

public class Client
{
public static void main(String a[])
{
Product b1=new Book();
Product b2=new Book();
Product a1=new Apple();
Visitor visitor;        BuyBasket basket=new BuyBasket();
        basket.addProduct(b1);
        basket.addProduct(b2);
        basket.addProduct(a1);        visitor=(Visitor)XMLUtil.getBean();        visitor.setName("张三");        basket.accept(visitor);
}
}

Customer.java

public class Customer extends Visitor
{
public void visit(Apple apple)
{
System.out.println("顾客" + name + "选苹果。");
}public void visit(Book book)
{
System.out.println("顾客" + name + "买书。");
}
}

Product.java

public interface Product
{
void accept(Visitor visitor);
}

Saler.java

public class Saler extends Visitor
{
public void visit(Apple apple)
{
System.out.println("收银员" + name + "给苹果过秤，然后计算其价格。");
}public void visit(Book book)
{
System.out.println("收银员" + name + "直接计算书的价格。");
}
}

Visitor.java

public abstract class Visitor
{
protected String name;public void setName(String name)
{
this.name=name;
}public abstract void visit(Apple apple);public abstract void visit(Book book);
}