问修改Python代码以便重复使用(Xlsx到Txt文件和WordCloud )

EN

import openpyxl as xl
import wordcloud
from wordcloud import WordCloud,STOPWORDS
from matplotlib.pyplot import imread
import matplotlib.pyplot as plt
import matplotlib.image as mpimg
import numpy as np

import jieba
import pandas as pd



def open_excel(filename):
    global wb1,ws1,filenametxt
    wb1 =xl.load_workbook(filename)
    ws1 = wb1.worksheets[0]
    filenametxt = filename
    print('Loading WorkBook Completed')
def create_sheet(filename1):
    global wb2,ws2
    wb2 = xl.load_workbook(filename1)
    wb2.create_sheet('Sheet2')
    ws2 = wb2.worksheets[1]
    print('Sheet 2 Created')

    mr = ws1.max_row
    mc = ws1.max_column
    minr = ws2.min_row

    for i in range(1, mr + 1):
        for j in range(0, mc + 1):
         # reading cell value from source excel file
            c = ws1.cell(row=i + 1, column=2)
            ws2.cell(row=i + 1, column=2).value = c.value
            wb2.save(filename1)
    print("Data Extracted To 'Column B'" )
    ws2.delete_cols(1)
    print('Empty Space in Column 1 Deleted')
    wb2.save(filename1)


def create_txtf(tfile):
    global df
    df = pd.read_excel(filenametxt, sheet_name=1)
    with open(tfile, mode='w', encoding='utf-8') as outfile:
        df.to_string(outfile, header=None, index=None)
        print('txt file created as file.txt')

def convert_remove(tfile1):
    #open read & remove spaces from txt file
    global commentfiletxt
    commentfiletxt = tfile1

    with open(commentfiletxt, 'r', encoding='utf-8') as f:
        lines = f.readlines()
    # # remove spaces
    lines = [line.replace(' ', '') for line in lines]
    print('Empty Spaces ' ' are removed' )
    # # finally, write lines in the file
    with open(commentfiletxt, 'w', encoding='utf-8') as f:
        f.writelines(lines)
        print('Data is written correctly without spaces')
    return(tfile1)


def wordcloudpic(picname,maskpathn):

    stopwords = set(STOPWORDS)
    stopwords.update(['此用户没有填写评论', 'hellip','zwj','其他特色','还没用','非常喜欢','产品功能','没有用','东西收到了','S','sode','c','s左右','u','middot','u','theta','rdquo','ldquo','ec','ok','好评','不错','很好','满意','好用','老板大气','好',\
                      'nbsp'])
    mask = imread(maskpathn)
    mask = mask.astype(np.uint8)

    with open(commentfiletxt, 'r',encoding='utf-8') as file:
        text = file.read()
        words = jieba.lcut(text)  # 精确分词
        newtxt = ' '.join(words)  # 空格拼接
        wd = wordcloud.WordCloud(stopwords=stopwords,\
                        font_path="MSYH.TTC",\
                        background_color="white", \
                        width=800, \
                        height=300, \
                        max_words=500, \
                        max_font_size=200, \
                        mask = mask, \
                        ).generate(text)

    txt = open(commentfiletxt, mode='r', encoding='utf-8')
    #  save picture
    wd.to_file(picname)


if __name__ == "__main__":
     open_excel("C:\\Users\\shakesmilk\\Desktop\\testtest\\test天猫商品评.xlsx")
     create_sheet("C:\\Users\\shakesmilk\\Desktop\\testtest\\test天猫商品评.xlsx")
     create_txtf("C:\\Users\\shakesmilk\\Desktop\\testtest\\file.txt")
     convert_remove("C:\\Users\\shakesmilk\\Desktop\\testtest\\file.txt")
     wordcloudpic('test.png','bubble.jpg')