32 lines
970 B
Python
32 lines
970 B
Python
import os
|
|
import sys
|
|
from aimodel import aimodel, aimodel_wrapper
|
|
from article_list import article, article_list
|
|
from fetch_exper_data import download_xml, read_xml_from_file, xml_dict_to_article_list
|
|
|
|
|
|
def main():
|
|
url_root = os.getenv("RSSHUB_ROOT")
|
|
assert url_root is not None
|
|
api_key = os.getenv("GEMINI_API_KEY")
|
|
assert api_key is not None
|
|
model = aimodel(api_key)
|
|
wrapper = aimodel_wrapper(model)
|
|
|
|
url = url_root + "/yicai/brief"
|
|
filename = "exper/raw/yicai.brief.xml"
|
|
download_xml(url=url, path=filename)
|
|
xml_dict = read_xml_from_file(filename)
|
|
article_list = xml_dict_to_article_list(
|
|
xml_dict, ["rss", "channel", "item"], ["title", "description"]
|
|
)
|
|
for i in range(0, len(article_list.list)):
|
|
article_list.list[i].replace_keywords(
|
|
wrapper.content_to_5_keywords(article_list.list[i].get_content(), "ZH")
|
|
)
|
|
print(str(article_list))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|