reykit 1.0.0__py3-none-any.whl → 1.0.1__py3-none-any.whl

reywechat/rwechat.py

@@ -0,0 +1,201 @@
+# !/usr/bin/env python
+# -*- coding: utf-8 -*-
+
+"""
+@Time    : 2023-10-17 20:27:16
+@Author  : Rey
+@Contact : reyxbo@163.com
+@Explain : WeChat methods.
+"""
+
+
+from typing import Literal, Optional, Union
+from os import getcwd as os_getcwd
+from os.path import join as os_join
+from reydb.rconnection import RDatabase as RRDatabase
+from reykit.ros import create_folder as reytool_create_folder
+from reykit.rsystem import block
+
+
+__all__ = (
+    'RWeChat',
+)
+
+
+class RWeChat(object):
+    """
+    Rey's `WeChat` type.
+
+    Will start client API service with port `19088` and message callback service with port '19089'.
+
+    Warnings, only applicable to WeChat clients with version `3.9.5.81`.
+
+    Warnings, must close file automatic download.
+
+    Warnings, the operating system version cannot be lower than `Windows 10 version 1709` or `Windows Server 2016 version 1709`.
+
+    Warnings, need support “Microsoft Visual C++2015”.
+    """
+
+
+    def __init__(
+        self,
+        rrdatabase: Optional[Union[RRDatabase, dict[Literal['wechat', 'file'], RRDatabase]]],
+        max_receiver: int = 2,
+        bandwidth_downstream: float = 5,
+        bandwidth_upstream: float = 5,
+        project_dir: Optional[str] = None
+    ) -> None:
+        """
+        Build `WeChat` attributes.
+
+        Parameters
+        ----------
+        rrdatabase : `RDatabase` instance of `reykit` package.
+            - `RDatabase`, Set all `RDatabase`: instances.
+            - `dict`, Set each `RDatabase`: instance, all item is required.
+                `Key 'wechat'`: `RDatabase` instance used in WeChat methods.
+                `Key 'file'`: `RDatabase` instance used in file methods.
+        max_receiver : Maximum number of receivers.
+        bandwidth_downstream : Download bandwidth, impact receive timeout, unit Mpbs.
+        bandwidth_upstream : Upload bandwidth, impact send interval, unit Mpbs.
+        project_dir: Project directory, will create sub folders.
+            - `None`: Use working directory.
+            - `str`: Use this directory.
+        """
+
+        # Import.
+        from .rclient import RClient
+        from .rdatabase import RDatabase
+        from .rlog import RLog
+        from .rreceive import RReceive
+        from .rschedule import RSchedule
+        from .rsend import RSend
+
+        # Create folder.
+        project_dir = project_dir or os_getcwd()
+        self._create_folder(project_dir)
+
+        # Set attribute.
+
+        ## Instance.
+        self.rclient = RClient(self)
+        self.rlog = RLog(self)
+        self.rreceive = RReceive(self, max_receiver, bandwidth_downstream)
+        self.rsend = RSend(self, bandwidth_upstream)
+        self.rdatabase = RDatabase(self, rrdatabase)
+        self.rschedule = RSchedule(self)
+
+        ## Client.
+        self.client_version = self.rclient.client_version
+        self.client_version_int = self.rclient.client_version_int
+        self.client_version_simulate = self.rclient.client_version_simulate
+        self.client_version_simulate_int = self.rclient.client_version_simulate_int
+        self.client_api_port = self.rclient.client_api_port
+        self.message_callback_port = self.rclient.message_callback_port
+
+        ## Receive.
+        self.receive_add_handler = self.rreceive.add_handler
+        self.receive_start = self.rreceive.start
+        self.receive_stop = self.rreceive.stop
+
+        ## Send.
+        self.send_add_handler = self.rsend.add_handler
+        self.send = self.rsend.send
+        self.send_start = self.rsend.start
+        self.send_stop = self.rsend.stop
+        self.wrap_try_send = self.rsend.wrap_try_send
+
+        ## Execute.
+        self.rexecute = self.rreceive.rexecute
+        self.execute_add_rule = self.rexecute.add_rule
+
+        ## Schedule.
+        self.schedule_add = self.rschedule.add
+        self.schedule_pause = self.rschedule.pause
+        self.schedule_resume = self.rschedule.resume
+
+
+    def _create_folder(
+        self,
+        project_dir: str
+    ) -> None:
+        """
+        Create project standard folders.
+
+        Parameters
+        ----------
+        project_dir: Project directory, will create sub folders.
+        """
+
+        # Set parameter.
+        folders = (
+            'Log',
+            'File'
+        )
+        folder_dict = {
+            folder: os_join(project_dir, folder)
+            for folder in folders
+        }
+
+        # Create.
+        paths = folder_dict.values()
+        reytool_create_folder(*paths)
+
+        # Set attribute.
+        self.dir_log = folder_dict['Log']
+        self.dir_file = folder_dict['File']
+
+
+    def start(self) -> None:
+        """
+        Start all methods.
+        """
+
+        # Start.
+        self.receive_start()
+        self.send_start()
+
+
+    def keep(self) -> None:
+        """
+        Blocking the main thread to keep running.
+        """
+
+        # Report.
+        print('Keep runing.')
+
+        # Blocking.
+        block()
+
+
+    @property
+    def print_colour(self) -> bool:
+        """
+        Whether print colour.
+
+        Returns
+        -------
+        Result.
+        """
+
+        # Get parameter.
+        result = self.rlog.rrlog.print_colour
+
+        return result
+
+
+    @print_colour.setter
+    def print_colour(self, value: bool) -> None:
+        """
+        Set whether print colour.
+
+        Parameters
+        ----------
+        value : Set value.
+        """
+
+        # Set.
+        self.rlog.rrlog.print_colour = value
+        self.rlog.rrlog_print.print_colour = value
+        self.rlog.rrlog_file.print_colour = value

reyworm/__init__.py

@@ -0,0 +1,24 @@
+# !/usr/bin/env python
+# -*- coding: utf-8 -*-
+
+"""
+@Time    : 2023-02-19 18:59:26
+@Author  : Rey
+@Contact : reyxbo@163.com
+@Explain : Rey's worm method set.
+
+Modules
+-------
+rall : All methods.
+rbrowser : Browser methods.
+rcalendar : Calendar methods.
+rnews : News methods.
+rsecurity : Security methods.
+rtranslate : Translate methods.
+"""
+
+
+from typing import Final
+
+
+__version__: Final[str] = '1.0.0'

reyworm/rall.py

@@ -0,0 +1,16 @@
+# !/usr/bin/env python
+# -*- coding: utf-8 -*-
+
+"""
+@Time    : 2022-12-08 13:11:09
+@Author  : Rey
+@Contact : reyxbo@163.com
+@Explain : All methods.
+"""
+
+
+from .rbrowser import *
+from .rcalendar import *
+from .rnews import *
+from .rsecurity import *
+from .rtranslate import *

reyworm/rbrowser.py

@@ -0,0 +1,134 @@
+# !/usr/bin/env python
+# -*- coding: utf-8 -*-
+
+"""
+@Time    : 2023-12-29 23:14:18
+@Author  : Rey
+@Contact : reyxbo@163.com
+@Explain : Browser methods.
+"""
+
+
+from typing import Any, Literal, Optional
+from selenium.webdriver import Edge, Chrome, EdgeOptions, ChromeOptions
+from reykit.rcomm import join_url
+from reykit.rtime import sleep
+
+
+__all__ = (
+    'RBrowser',
+    'get_page'
+)
+
+
+class RBrowser(object):
+    """
+    Rey's `browser` type.
+    """
+
+
+    def __init__(
+        self,
+        driver: Literal['edge', 'chrome'] = 'edge',
+        headless: bool = False
+    ) -> None:
+        """
+        Build `browser` attributes.
+
+        Parameters
+        ----------
+        driver : Browser driver type.
+            - `Literal['edge']`: Edge browser.
+            - `Literal['chrome']`: Chrome browser.
+        headless : Whether use headless mode.
+        """
+
+        # Get parameter.
+        match driver:
+            case 'edge':
+                driver_type = Edge
+                driver_option_type = EdgeOptions
+            case 'chrome':
+                driver_type = Chrome
+                driver_option_type = ChromeOptions
+
+        # Option.
+        options = driver_option_type()
+
+        ## Headless.
+        if headless:
+            options.add_argument('--headless')
+
+        # Driver.
+        self.driver = driver_type(options)
+
+
+    def request(
+        self,
+        url: str,
+        params: Optional[dict[str, Any]] = None
+    ) -> None:
+        """
+        Request URL.
+
+        Parameters
+        ----------
+        url : URL.
+        params : URL parameters.
+        """
+
+        # Get parameter.
+        if params is None:
+            params = {}
+        url = join_url(url, params)
+
+        # Request.
+        self.driver.get(url)
+
+
+    @property
+    def page(self) -> str:
+        """
+        Return page elements document.
+
+        Returns
+        -------
+        Page elements document.
+        """
+
+        # Get parameter.
+        page_source = self.driver.page_source
+
+        return page_source
+
+
+    __call__ = request
+
+
+def get_page(
+    url: str,
+    params: Optional[dict[str, Any]] = None
+) -> str:
+    """
+    Get page elements document.
+
+    Parameters
+    ----------
+    url : URL.
+    params : URL parameters.
+
+    Returns
+    -------
+    Page elements document.
+    """
+
+    # Get parameter.
+    browser = RBrowser(headless=True)
+
+    # Request.
+    browser.request(url, params)
+
+    # Page.
+    page = browser.page
+
+    return page

reyworm/rcalendar.py

@@ -0,0 +1,159 @@
+# !/usr/bin/env python
+# -*- coding: utf-8 -*-
+
+"""
+@Time    : 2024-01-10 21:57:08
+@Author  : Rey
+@Contact : reyxbo@163.com
+@Explain : Calendar methods.
+"""
+
+
+from typing import Optional
+from json import loads as json_loads
+from reykit.rcomm import request
+from reykit.rregex import search
+from reykit.rtime import now
+
+
+__all__ = (
+    'get_calendar',
+    'get_lunar_calendar'
+)
+
+
+def get_calendar(
+    year: Optional[int] = None,
+    month: Optional[int] = None
+) -> list[dict]:
+    """
+    Get calendar table for three months from `baidu` website.
+
+    Parameters
+    ----------
+    year : Given year.
+        - `None`: Now year.
+    month : Given month.
+        - `None`: Now month.
+
+    Returns
+    -------
+    Calendar table.
+    """
+
+    # Get parameter.
+    now_date = now('date')
+    year = year or now_date.year
+    month = month or now_date.month
+    if month == 12:
+        month = 1
+    else:
+        month += 1
+    url = 'https://opendata.baidu.com/data/inner'
+    query = '%s年%s月' % (year, month)
+    params = {
+        'tn': 'reserved_all_res_tn',
+        'type': 'json',
+        'resource_id': '52109',
+        'query': query,
+        'apiType': 'yearMonthData',
+        'cb': 'jsonp_1706670926975_94318'
+    }
+
+    # Request.
+    response = request(url, params)
+
+    # Extract.
+    pattern = '{.+}'
+    text = search(pattern, response.text)
+    data: dict = json_loads(text)
+    table: list[dict] = data['Result'][0]['DisplayData']['resultData']['tplData']['data']['almanac']
+
+    # Convert.
+    week_dict = {
+        '一': 0,
+        '二': 1,
+        '三': 2,
+        '四': 3,
+        '五': 4,
+        '六': 5,
+        '日': 6
+    }
+    table = [
+        {
+            'year': int(row['year']),
+            'month': int(row['month']),
+            'day': int(row['day']),
+            'week': week_dict[row['cnDay']],
+            'work': row.get('status'),
+            'festival': [
+                {
+                    'name': info['name'],
+                    'url': info.get('baikeUrl')
+                }
+                for info in row.get('festivalInfoList', [])
+            ],
+            'animal': row['animal'],
+            'lunar_year': int(row['lunarYear']),
+            'lunar_month': int(row['lunarMonth']),
+            'lunar_day': int(row['lunarDate']),
+            'gz_year': row['gzYear'],
+            'gz_month': row['gzMonth'],
+            'gz_day': row['gzDate'],
+            'suit': row['suit'].split('.'),
+            'avoid': row['avoid'].split('.'),
+            'url': row['yjJumpUrl']
+        }
+        for row in table
+    ]
+    for row in table:
+        week = row['week']
+        work = row['work']
+        match work:
+            case None:
+                is_work_day = week not in (5, 6)
+            case '1':
+                is_work_day = False
+            case '2':
+                is_work_day = True
+        row['work'] = is_work_day
+
+    return table
+
+
+def get_lunar_calendar(
+    year: Optional[int] = None,
+    month: Optional[int] = None
+) -> list[dict]:
+    """
+    Get lunar calendar table for one month from `rili` website.
+
+    Parameters
+    ----------
+    year : Given year.
+        - `None`: Now year.
+    month : Given month.
+        - `None`: Now month.
+
+    Returns
+    -------
+    Lunar calendar table.
+    """
+
+    # Get parameter.
+    now_date = now('date')
+    year = year or now_date.year
+    month = month or now_date.month
+    url = 'https://www.rili.com.cn/rili/json/pc_wnl/%s/%02d.js' % (year, month)
+    params = {'_': now('timestamp')}
+
+    # Request.
+    response = request(url, params)
+
+    # Extract.
+    pattern = '{.+}'
+    text = search(pattern, response.text)
+    data = json_loads(text)
+    table = data['data']
+
+    return table

reyworm/rnews.py

@@ -0,0 +1,126 @@
+# !/usr/bin/env python
+# -*- coding: utf-8 -*-
+
+"""
+@Time    : 2024-01-22 14:06:05
+@Author  : Rey
+@Contact : reyxbo@163.com
+@Explain : News methods.
+"""
+
+
+from typing import Any, Literal
+from reykit.rcomm import request, join_url
+from reykit.rtime import to_time, time_to
+
+
+__all__ = (
+    'get_weibo_hot_search',
+    'get_toutiao_hot_search'
+)
+
+
+def get_weibo_hot_search() -> list[dict[Literal['rank', 'time', 'title', 'type', 'hot', 'url'], Any]]:
+    """
+    Get hot search table from `weibo` website.
+
+    Returns
+    -------
+    Hot search table.
+        - `Key 'rank'`: Hot search rank.
+        - `Key 'time'`: Hot search time.
+        - `Key 'title'`: Hot search title.
+        - `Key 'type'`: Hot search type.
+        - `Key 'hot'`: Hot search hot value.
+        - `Key 'url'`: Hot search URL.
+    """
+
+    # Request.
+    url = 'https://weibo.com/ajax/side/hotSearch'
+    response = request(url, check=True)
+
+    # Extract.
+    response_json = response.json()
+    table: list[dict] = response_json['data']['realtime']
+
+    # Convert.
+    table = [
+        {
+            'title': info['word'],
+            'hot': info['num'],
+            'url': join_url(
+                'https://s.weibo.com/weibo',
+                {'q': '#%s#' % info['word']}
+            )
+        }
+        for info in table
+        if 'flag' in info
+    ]
+    func_sort = lambda row: (
+        0
+        if row['hot'] is None
+        else row['hot']
+    )
+    table.sort(key=func_sort, reverse=True)
+    table = [
+        {
+            'rank': index,
+            **row
+        }
+        for index, row in enumerate(table)
+    ]
+
+    return table
+
+
+def get_toutiao_hot_search() -> list[dict[Literal['title', 'type', 'label', 'hot', 'url', 'image'], Any]]:
+    """
+    Get hot search table from `toutiao` website.
+
+    Returns
+    -------
+    Hot search table.
+        - `Key 'title'`: Hot search title.
+        - `Key 'type'`: Hot search type list.
+        - `Key 'label'`: Hot search label.
+        - `Key 'hot'`: Hot search hot value.
+        - `Key 'url'`: Hot search URL.
+        - `Key 'image'`: Hot search image URL.
+    """
+
+    # Request.
+    url = 'https://www.toutiao.com/hot-event/hot-board/'
+    params = {'origin': 'toutiao_pc'}
+    response = request(
+        url,
+        params,
+        check=True
+    )
+
+    # Extract.
+    response_json = response.json()
+    table: list[dict] = response_json['data']
+
+    # Convert.
+    table = [
+        {
+            'title': info['Title'],
+            'type': info.get('InterestCategory'),
+            'label': info.get('LabelDesc'),
+            'hot': int(info['HotValue']),
+            'url': info['Url'],
+            'image': info['Image']['url'],
+        }
+        for info in table
+    ]
+    func_sort = lambda row: row['hot']
+    table.sort(key=func_sort, reverse=True)
+    table = [
+        {
+            'rank': index,
+            **row
+        }
+        for index, row in enumerate(table)
+    ]
+
+    return table