사용자:하늘/양식

ceruleanbot

https://www.geeksforgeeks.org/how-to-use-html-in-tkinter-python/

import requests
S = requests.Session()


class wiki:
    def __init__(self, URL: str):
        self.url = URL
        self.csrftoken = False
        self.pages = {}
        self.page = None

    def addpage(self, __page__=None):  # add page to page list #1 param is object or str
        if type(__page__) == str:
            self.pages[__page__] = page(self.url, __page__, self.csrftoken)
        elif type(__page__) == object:
            self.pages[__page__.title] = __page__  # elif type(page) == dict:
        elif __page__ == None:					# setpage() 호출 후에 addpage() 호출
            if self.page != None:
                self.pages[self.page.title] = self.page

    def setpage(self, pagename: str):  # 사이트 객체 내에 페이지 객체 저장, 리턴
        self.page = page(self.url, pagename, self.csrftoken)  # 내부 사이트 객체에 저장
        return self.page  # 페이지 객체 리턴, 사용법 : a = wiki.page('asdf') a.get()

    def get_csrftoken(self):
        params = {"action": "query", "meta": "tokens", "format": "json"}
        DATA = S.get(url=self.url, params=params).json()
        self.csrftoken = DATA['query']['tokens']['csrftoken']

    def login(self, name: str, password: str):
        # Step 1: Retrieve login token first
        params = {
            'action': "query",
            'meta': "tokens",
            'type': "login",
            'format': "json"
        }
        DATA = S.get(url=self.url, params=params).json()
        LOGIN_TOKEN = DATA['query']['tokens']['logintoken']
        # Step 2: Send a POST request to login. Using the main account for login is not
        # supported. Obtain credentials via Special:BotPasswords
        # 봇 비밀번호 설정 권장, 형식 : Caeruleum@asdfqwerzxcvpoiulkjhmnbv
        params = {
            "action": "login",
            "format": "json",
            "lgname": name,
            "lgpassword": password,
            "lgtoken": LOGIN_TOKEN,
            "utf8": 1
        }
        DATA = S.post(self.url, data=params).json()
        if DATA["login"]["result"] == "Success":
            print("로그인 성공")
            self.get_csrftoken()
        else:
            print("로그인 실패")
            exit()

    def logout(self):
        DATA = S.post(self.url, data={"action": "logout",
                                      "token": self.csrftoken, "format": "json"}).json()
        print("로그아웃")
        print(DATA)

    def pagesInCategory(self, category: str, namespace: str = ''):
        params = {
            "action": "query",
            "format": "json",
            "list": "categorymembers",
            "utf8": 1,
            "cmnamespace": namespace,
            "cmtitle": 'category:'+category,
            "cmprop": "title|timestamp",
            "cmlimit": "max",
            "cmsort": "timestamp",
        }
        data = S.get(url=self.url, params=params).json()
        return data['query']['categorymembers']

    def savePages(self, summary: str = ''):  # 작업 목록의 문서들 모두 저장. wiki.pages
        for __page__ in self.pages:
            __page__.save(summary)

    def savePage(self, summary: str = ''):  # 사이트 객체 내의 유일 페이지 객체를 위키에 업로드. wiki.page
        self.page.save(summary)

    def removecategory(self):
        ''


class page:
    def __init__(self, url, title, tokens):
        self.url = url
        self.title = title
        self.csrftoken = tokens

    def get(self):
        params = {
            "action": "query",
            "format": "json",
            "prop": "revisions",
            "formatversion": "2",
            "rvprop": "content",
            "rvslots": "*",
            "titles": self.title
        }
        pageData = S.get(url=self.url, params=params).json()[
            'query']['pages'][0]
        if not 'missing' in pageData:
            self.cont = pageData['revisions'][0]['slots']['main']['content']
        else:
            self.cont = 'missing page'

    def print(self):
        print(self.cont)

    def remove(self, toremove: str):
        self.cont = self.cont.replace(toremove, '')

    def replace(self, toreplace: str, replace: str) -> None:
        self.cont = self.cont.replace(toreplace, replace)

    def save(self, summary: str = ''):
        if self.csrftoken == False:
            print('csrftoken is ungetted')
            return
        print(self.title+' 편집')
        params = {
            "action": "edit",
            "format": "json",
            "title": self.title,
            "text": self.cont,
            "summary": summary,
            "token": self.csrftoken,
            "utf8": 1,
            "formatversion": "latest",
            "bot": "true"
        }
        DATA = S.post(url=self.url, data=params).json()
        print(DATA)

위의 봇과 통합 예정 GUI로도 CLI로도 사용할 수 있게

import ceruleanbot
import time
import requests
import tkinter

window = tkinter.Tk()
window.title("cerulwikibot")

S = requests.Session()
libre = cerulwikibot.wiki('https://librewiki.net/api.php')

def searchdoc():
	title = entry.get()
	if title == '' :
		label.configure(text='?')
		text.delete(1.0, tkinter.END)
		text.insert(1.0, '?')
	else:
		label.configure(text=title)
		p = libre.page(title)
		p.get()
		text.delete(1.0, tkinter.END)
		text.insert(1.0, p.cont)

label = tkinter.Label(window, text = '문서 제목 입력하고 소스보기', height=2)
label.pack()

text = tkinter.Text(window)
text.focus()
text.pack()

entry = tkinter.Entry(window, width = 50)
entry.pack()

viewsource = tkinter.Button(window, text='소스 보기', command=searchdoc)
viewsource.pack(pady=0)

savebutton = tkinter.Button(window, text='편집 저장', command=searchdoc) #미구현
savebutton.pack(pady=0)

window.mainloop()