第一步，拆分規(guī)則

把規(guī)則部分的代碼，復制到新文件：rule.py，原來的代碼注釋或刪除。這里為了對比，DE8UG 在 08 號文件進行了注釋，見最后的代碼。

# rule.py
# author: DE8UG
# PROJECT: rengleme

rule = {
   "濕垃圾": ["菜葉", "橙皮", "蔥", "餅干"],
   "干垃圾": ["舊浴缸", "盆子", "海綿", "衛(wèi)生紙"]
}

第二步，拆分處理垃圾的核心代碼

新建文件：handle_laji.py，把處理垃圾分類的函數(shù)復制到這里。

# handle_laji.py
# author: DE8UG
# PROJECT: rengleme

def reng_laji(rule_k, laji, ljt):
   if rule_k == ljt['name']:
       ljt['data'].append(laji)


def fenlei(laji, rule, ljt):
   for k, v in rule.items():
       print(k, v)
       if laji in v:
           print('找到了垃圾：', laji, k)
           reng_laji(k, laji, ljt)

第三步，處理文件，加載和保存垃圾

新建文件：handle_file.py，把處理文件的函數(shù)復制到這里。

# handle_file.py
# author: DE8UG
# PROJECT: rengleme

import json


# print('加載文件：', ljt_gan)
def load_data(filename):
   with open(filename) as f:
       data = json.load(f)
       return data


# 用函數(shù)封裝寫文件的代碼
def save_to_file(filename, data):
   with open(filename, 'w') as f:
       json.dump(data, f)

第四步，完善代碼，測試

把 08 文件的已經(jīng)移除的代碼刪除（我這里為了對比僅做了注釋），然后在最前面導入前面幾個新 py 文件。

注意：這幾個文件需要在一個目錄下，以免出現(xiàn)不同路徑找不到文件的情況。

import 導入相應的模塊后，后續(xù)的函數(shù)，需要用模塊.函數(shù)名的方式調(diào)用。請自行補充所需函數(shù)名。然后進行代碼測試。沒有問題的話，我們今天新的改造就完成了。

# author: DE8UG

# 創(chuàng)建垃圾桶和垃圾
# 規(guī)則：k:v,一個k對應一個v，代碼表示用冒號分割
# 輸入
# 判斷
# 保存
# 函數(shù)
# 綜合
# 8 劃分模塊，轉移代碼


# import json

# 導入自建模塊
import rule
import handle_file, handle_laji

# rule = {
#     "濕垃圾": ["菜葉", "橙皮", "蔥", "餅干"],
#     "干垃圾": ["舊浴缸", "盆子", "海綿", "衛(wèi)生紙"]
# }


# # print('加載文件：', ljt_gan)
# def load_data(filename):
#     with open(filename) as f:
#         data = json.load(f)
#         return data


# def reng_laji(rule_k, laji, ljt):
#     if rule_k == ljt['name']:
#         ljt['data'].append(laji)


# def fenlei(laji, rule, ljt):
#     for k, v in rule.items():
#         print(k, v)
#         if laji in v:
#             print('找到了垃圾：', laji, k)
#             reng_laji(k, laji, ljt)
           # reng_laji(k, laji, ljt_shi)


# 用函數(shù)封裝寫文件的代碼
# def save_to_file(filename, data):
#     with open(filename, 'w') as f:
#         json.dump(data, f)


def main():

   # 定義垃圾桶
   ljt_shi = {
       'name': "濕垃圾",
       'data': []
   }

   ljt_gan = {
       'name': "干垃圾",
       'data': []
   }

   # 加載已有的垃圾
   ljt_gan = handle_file.load_data('gan.json')
   ljt_shi = handle_file.load_data('shi.json')

   print('加載文件：', ljt_gan, ljt_shi)

   # 扔垃圾
   laji = input("輸入要扔的垃圾：")
   print("垃圾：", laji)

   # 分類
   handle_laji.fenlei(laji, rule, ljt_gan)
   handle_laji.fenlei(laji, rule, ljt_shi)

   print('-'*20)
   print(ljt_shi)
   print(ljt_gan)

   # 調(diào)用函數(shù)，保存到具體的垃圾桶文件
   handle_file.save_to_file('gan.json', ljt_gan)
   handle_file.save_to_file('shi.json', ljt_shi)


if __name__ == "__main__":
   main()