def task_category(logger, is_test=False):
sql_connect = Table(logger=logger)
sql_connect.create_table_category()
category = Category(logger=logger)
category_info = category.get_category()
for each in category_info:
title = each.get('title')
url = each.get('url')
category = putty(title)
if is_test:
message = json.dumps(dict(category=category, url=url),
ensure_ascii=False)
logger.debug(message)
else:
sql_connect.insert_table_category(category=category, url=url)
sql_connect.close()
def set_category(self):
cat = Category(self.text)
self.category = cat.get_category()
