def load_seed_keywords(keywords_file: str) -> Category:
"""
获取初始关键词, 并构建分类树
:param keywords_file: 关键词文件路径
:return: 分类树root节点
"""
category = Category("ROOT")
category_list = []
with open(keywords_file, "r", encoding="utf-8") as fr:
for line in fr:
categories, words = line.strip().split("###")
category_list.append(categories)
categories = categories.split("/")
words = set(words.split("|"))
category.add_category(categories).set_keywords(words)
category.set_category_list(category_list)
logging.info("分类树: {}".format(category))
return category
def new_category():
print("-"*30)
print("Please Enter New Category Name : ")
new_cat = input("Category : ")
ref_var = db.reference('Category').order_by_child('category_name').equal_to(new_cat).get()
if ref_var:
return False
category_object = Category()
new_category_id = category_object.add_category(new_cat)
return new_cat
from parse_money_tracker_data import MoneyTrackerData
from aggregated_money_tracker import AggregatedMoneyTracker
from category import Category
data_obj = MoneyTrackerData('money_tracker.txt')
data_list = data_obj.list_user_data()
# print(data_list)
agr_data = AggregatedMoneyTracker(data_list)
agr_dict = agr_data.aggregate_data()
# print(agr_data._dict_data())
print(agr_dict)
income = Category('income', agr_dict)
# print(income.get_data_from_category())
income.add_category('31-03-2018', 'income', agr_dict)
income.add_category('31-03-2018', 'expense', agr_dict)
income.add_category('30-03-2018', 'expense', agr_dict)
print(income.get_data_from_category())
print(agr_dict)
