Files
peter/seek/ofweek_com/ai.py
konjacpotato 8c1a740f0b import peter
2025-11-12 20:42:16 +08:00

63 lines
2.3 KiB
Python

import datetime
from DrissionPage.errors import ElementNotFoundError
from database.tinformationsource.model import TInformationSource
from database.tnews.model import TNews
from log.log_manager import logger
from seek.seek_base import SeekBase
from utils.time_utils import process_time
class OfweekComAi(SeekBase):
def __init__(self, information_source: TInformationSource):
super().__init__(information_source)
def get_news(self):
news_result = []
_news_list = self.session.s_ele('.main-cont-left w640').s_eles('.^top-title')
for _news in _news_list:
try:
__news = TNews()
__news.title = _news.s_ele('tag:a').text
__news.url = _news.s_ele('tag:a').link
_time = _news.parent().s_eles('tag:span')[4].text
__news.occurrence_date = process_time(_time)
__news.source = self.information_source.title
news_result.append(__news)
except ElementNotFoundError as e:
logger.error(f"ElementNotFoundError: {e} - Failed to find element in news item.")
except Exception as e:
logger.error(f'Unexpected error occurred: {e}')
return news_result
def get_news(information_source: TInformationSource) -> list:
ofweek_com_ai = OfweekComAi(information_source)
news_list = ofweek_com_ai.get_news()
ofweek_com_ai.finish()
return news_list
def news_task(information_source: TInformationSource):
logger.info(f'{information_source.title} news_task start execute at {datetime.datetime.now()}', )
ofweek_com_ai = OfweekComAi(information_source)
ofweek_com_ai.do_seek_task()
ofweek_com_ai.finish()
logger.info(f'{information_source.title} news_task end execute at {datetime.datetime.now()}')
if __name__ == '__main__':
logger.info('This module is not for direct call!')
information_source_ = TInformationSource()
information_source_.is_static = True
information_source_.url = 'https://www.ofweek.com/ai/'
information_source_.title = '人工智能_维科网'
news_task(information_source_)
# news_list_ = get_news(information_source_)
# for news in news_list_:
# print(news)
logger.info('Done.')