WikiBot.py

import datetime
import os

from config import (AKASHI_LIST_OUTPUT_LUA, ITEMS_DATA, ITEMS_DATA_V2, LUATABLE_PATH,
                    OUPUT_PATH, SHINKAI_ITEMS_DATA, SHINKAI_SHIPS_DATA,
                    SHIPS_DATA, TIMEZONE, WIKIWIKI_Compare_TABLE, SHIP_SERIES_DB,
                    WIKIWIKI_MaxValue_TABLE, AIRPOWER_TABLE, SHIPCLASSES_MAPPING_DATA)
from HttpClient import HttpClient


class KcwikiException(Exception):
    def __init__(self, message):
        super().__init__(message)


class WikiBot(HttpClient):

    KCWIKI_API_URL = 'http://zh.kcwiki.cn:8000/api.php'
    HEADERS = {
        'Cache-Control': 'no-cache',
        'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/64.0.3282.140 Safari/537.36'
    }

    def __init__(self, username, password):
        super().__init__()
        self.loginToken = ''
        self.editToken = ''
        self.username = username
        self.password = password

    async def login(self):
        rdata = {
            'action': 'query',
            'meta': 'tokens',
            'type': 'login',
            'format': 'json'
        }
        async with self.session.post(self.KCWIKI_API_URL, data=rdata) as resp:
            try:
                resp_json = await resp.json()
                if not resp_json:
                    raise KcwikiException('Wiki-Bot: Failed to get login token!')
                self.loginToken = resp_json['query']['tokens']['logintoken']
            except Exception:
                raise KcwikiException('Wiki-Bot: Failed to get login token!')
        rdata = {
            'action': 'login',
            'format': 'json',
            'lgtoken': self.loginToken
        }
        rdata['lgname'] = self.username
        rdata['lgpassword'] = self.password
        async with self.session.post(self.KCWIKI_API_URL, data=rdata) as resp:
            try:
                resp_json = await resp.json()
                if not resp_json:
                    raise KcwikiException('Wiki-Bot: Failed to log in!')
                if resp_json['login']['result'] == 'Failed':
                    raise KcwikiException(resp_json['login']['reason'])
            except Exception:
                raise KcwikiException('Wiki-Bot: Failed to log in!')
        rdata = {
            'action': 'query',
            'meta': 'tokens',
            'format': 'json'
        }
        async with self.session.post(self.KCWIKI_API_URL, data=rdata) as resp:
            try:
                resp_json = await resp.json()
                if not resp_json:
                    raise KcwikiException('Wiki-Bot: Failed to get edit token!')
                self.editToken = resp_json['query']['tokens']['csrftoken']
            except Exception:
                raise KcwikiException('Wiki-Bot: Failed to get edit token!')

        if self.editToken == '+\\':
            raise KcwikiException('Wiki-Bot: Incorrect edittoken \'+\\\' !')
        print('Wiki-Bot: Login Successfully!')

    async def updatePage(self, page_title, filename):
        now = datetime.datetime.now(TIMEZONE)
        comment = 'KcWiki CI Robot: Update {}'.format(now.strftime('%Y-%m-%d %H:%M:%S'))
        rdata = {
            'action': 'edit',
            'title': page_title,
            'token': self.editToken,
            'summary': comment,
            'format': 'json'
        }
        if not os.path.exists(filename):
            print('Wiki-Bot: Page {{{{{}}}}} file {} not found, skip'.format(page_title, filename))
            return
        with open(filename, 'r', encoding='utf-8') as fp:
            text = fp.read()
            rdata.update({
                'text': text
            })
        async with self.session.post(self.KCWIKI_API_URL, data=rdata) as resp:
            try:
                resp_json = await resp.json()
                if resp_json['edit']['result'] != 'Success':
                    raise KcwikiException('Wiki-Bot: Failed to update page!')
            except KeyError:
                print(resp_json)
                print('Wiki-Bot: Page {{{{{}}}}} failed to update! (Kcwiki BOOM)'.format(page_title))
            except Exception:
                raise KcwikiException('Wiki-Bot: Failed to update page!')
        print('Wiki-Bot: Page {{{{{}}}}} updated successfully'.format(page_title))

    async def start(self):
        await self.login()
        await self.updatePage('模块:深海栖舰制空数据', OUPUT_PATH + LUATABLE_PATH + AIRPOWER_TABLE)
        await self.updatePage('模块:舰娘数据', OUPUT_PATH + LUATABLE_PATH + SHIPS_DATA + '.lua')
        await self.updatePage('模块:舰娘装备数据改', OUPUT_PATH + LUATABLE_PATH + ITEMS_DATA + '.lua')
        await self.updatePage('模块:舰娘装备数据改二', OUPUT_PATH + LUATABLE_PATH + ITEMS_DATA_V2 + '.lua')
        await self.updatePage('模块:舰娘系列数据', OUPUT_PATH + LUATABLE_PATH + SHIP_SERIES_DB + '.lua')
        await self.updatePage('模块:深海装备数据', OUPUT_PATH + LUATABLE_PATH + SHINKAI_ITEMS_DATA + '.lua')
        await self.updatePage('模块:深海栖舰数据改二', OUPUT_PATH + LUATABLE_PATH + SHINKAI_SHIPS_DATA + '.lua')
        await self.updatePage('模块:明石工厂数据', OUPUT_PATH + LUATABLE_PATH + AKASHI_LIST_OUTPUT_LUA)
        await self.updatePage('模块:ShipClass', OUPUT_PATH + LUATABLE_PATH + SHIPCLASSES_MAPPING_DATA + '.lua')
        await self.updatePage('User:MZI/装備の比較', OUPUT_PATH + WIKIWIKI_Compare_TABLE)
        await self.updatePage('User:MZI/最大值表', OUPUT_PATH + WIKIWIKI_MaxValue_TABLE)