"""
清洗Seed库，解决协议问题
"""

import re

from base.utils import mongo_helper
from base.utils.es_helper import es_query
from base.utils.mongo_helper import query_list_new


def change(item):
    re_news = item.get("re_news")
    print(re_news)
    re_news_new = re_news.replace("http:", "(http|https):").replace("https:", "(http|https):")
    print(re_news_new)
    mongo_helper.update_one(table="aic_news_range_seed", filter={"_id": item['_id']},
                            update={'$set': {"re_news": re_news_new}})


if __name__ == '__main__':
    items = query_list_new("aic_news_range_seed")
    for item in items:
        re_news = item.get("re_news")
        if re_news is not None and ("http:" in re_news or "https" in re_news):
            re_news_new = change(item)