# author: DE8UG
# 创建垃圾桶和垃圾
# 规则：k:v,一个k对应一个v，代码表示用冒号分割
# 输入
# 判断
# 保存
# 函数
# 综合


import json


rule = {
   "湿垃圾": ["菜叶", "橙皮", "葱", "饼干"],
   "干垃圾": ["旧浴缸", "盆子", "海绵", "卫生纸"]
}


# 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)


# 用函数封装写文件的代码
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 = load_data('gan.json')
   ljt_shi = load_data('shi.json')

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

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

   # 分类
   fenlei(laji, rule, ljt_gan)
   fenlei(laji, rule, ljt_shi)

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

   # 调用函数，保存到具体的垃圾桶文件
   save_to_file('gan.json', ljt_gan)
   save_to_file('shi.json', ljt_shi)


if __name__ == "__main__":
   main()

# Python算个球
# author：DE8UG
# 首先准备好目标文件
filename = 'rengleme_07.py'

# 写函数得到里面所有行
def get_lines(filename):
   with open(filename) as f:
       return f.readlines()


# 执行这个函数打印的话，会是下面这一大堆字符串，但，注意看，是个列表。那就好办了
"""
['# author: DE8UG\n', '# 创建垃圾桶和垃圾\n',
'# 规则：k:v,一个k对应一个v，代码表示用冒号分割\n',
'# 输入\n', '# 判断\n', '# 保存\n', '# 函数\n', '# 综合\n', '\n', '\n',

'import json\n', '\n', '\n',
   'rule = {\n', '    "湿垃圾": ["菜叶", "橙皮", "葱", "饼干"],
       \n', '
        "干垃圾": ["旧浴缸", "盆子", "海绵", "卫生纸"]\n', '}\n',
             '\n', '\n',
                 "# print('加载文件：', ljt_gan)\n",
'def load_data(filename):\n',
   '    with open(filename) as f:\n',
       '        data = json.load(f)\n',
           '        return data\n', '\n', '\n', 'def reng_laji(rule_k, laji, ljt):\n', "    if rule_k == ljt['name']:\n", "        ljt['data'].append(laji)\n", '\n', '\n', 'def fenlei(laji, rule, ljt):\n', '    for k, v in rule.items():\n', '        print(k, v)\n', '        if laji in v:\n', "            print('找到了垃圾：', laji, k)\n", '            reng_laji(k, laji, ljt)\n', '            # reng_laji(k, laji, ljt_shi)\n', '        \n', '\n', '# 用函数封装写文件的代码\n', 'def save_to_file(filename, data):\n', "    with open(filename, 'w') as f:\n", '        json.dump(data, f)\n', '\n', '\n', 'def main():\n', '\n', '    # 定义垃圾桶\n', '    ljt_shi = {\n', '        \'name\': "湿垃圾",\n', "        'data': []\n", '    }\n', '\n', '    ljt_gan = {\n', '        \'name\': "干垃圾",\n', "        'data': []\n", '    }\n', '\n', '    # 加载已有的垃圾\n', "    ljt_gan = load_data('gan.json')\n", "    ljt_shi = load_data('shi.json')\n", '\n', "    print('加载文件：', ljt_gan, ljt_shi)\n", '\n', '    # 扔垃圾\n', '    laji = input("输入要扔的垃圾：")\n', '    print("垃圾：", laji)\n', '\n', '    # 分类\n', '    fenlei(laji, rule, ljt_gan)\n', '    fenlei(laji, rule, ljt_shi)\n', '\n', "    print('-'*20)\n", '    print(ljt_shi)\n', '    print(ljt_gan)\n', '\n', '    # 调用函数，保存到具体的垃圾桶文件\n', "    save_to_file('gan.json', ljt_gan)\n", "    save_to_file('shi.json', ljt_shi)\n", '\n', '\n', 'if __name__ == "__main__":\n', '    main()\n', '\n', '\n', '\n']
"""

# 我们定义一下结果格式，把每一行代码格式化一下为字典，包含行号，字符串，类型，含义四部分，然后组合为新的列表
[
{
   'line':1,
   'str':'def load_data(filename):\n',
   'type':'function',   # code, comment, import, var
   'mean':'函数load_data，包含参数：filename'
}
]

def analysis_code(lines):
   """分析每一行代码，根据不同特点，划分不同类型，注意换行的变量和函数会有内部代码，我们通过上一次保存的结果进行分解判断。把格式化后的行保存为新列表，并返回。
   """
   results = []
   count = 1
   for line in lines:
       format_line = {
           'line': count,
           'str': line,
           'type':'code',   # code, comment, import, var
           'inner_code': [],
           'mean':''
       }
       if line.startswith('#'):
           format_line['type'] = 'comment'
       elif line.startswith('import'):
           format_line['type'] = 'import'
       elif line.find('=') > 0:
           format_line['type'] = 'var'
       elif line.startswith('def'):
           format_line['type'] = 'function'

       # 空行，函数内部代码，变量内部代码，不加入结果
       if line.startswith(' ') or line.startswith('}\n'):
           # print(results[-1])
           if results[-1]['type'] == 'function' or results[-1]['type'] == 'var':
               results[-1]['inner_code'].append(line)
       elif line.startswith('\n'):
           pass  # 空行直接掠过
       else:
           results.append(format_line)

       # 循环下一行
       count += 1

   return results


# 下面开始测试
if __name__ == "__main__":
   print('*'*50)
   print('欢迎使用Python算个球，掐指一算，你的代码内容如下：')
   lines = get_lines(filename)
   # print(lines[:10])
   r = analysis_code(lines)
   # print(r)
   comment_list = []
   function_list = []
   for line in r:
       # print(line)
       if line['type']=='comment':
           comment_list.append(line)
       elif line['type']=='function':
           function_list.append(line)
   print(f'- 共{len(comment_list)}个注释')
   print(f'- 共{len(function_list)}个函数：')
   for f in function_list:
       print(f'line:{f["line"]}', f['str'].strip())