This commit is contained in:
qcminecraft 2021-02-03 11:56:54 +08:00
parent e6251155da
commit 65fa432290
2 changed files with 38 additions and 3 deletions

1
README.md Normal file
View File

@ -0,0 +1 @@
测试

View File

@ -7,6 +7,9 @@ import requests
import xmltodict import xmltodict
import time import time
time_start = time.time()
print(
''' __ __ _ _ __ __ _ _ \n | \/ (_) | | / //_ | | | | \n | \ / |_ ___ ___| | _____ _ _ __ __/ /_ | | |__ ___ | |_ \n | |\/| | / __/ __| |/ / _ \ | | | \ \/ / '_ \| | '_ \ / _ \| __|\n | | | | \__ \__ \ < __/ |_| | > <| (_) | | |_) | (_) | |_ \n |_| |_|_|___/___/_|\_\___|\__, | /_/\_\\___/|_|_.__/ \___/ \__|\n __/ | \n |___/ ''')
config_file = open('config.json', 'r') config_file = open('config.json', 'r')
config = json.loads(config_file.read()) config = json.loads(config_file.read())
config_file.close() config_file.close()
@ -19,18 +22,49 @@ def xml_to_json(xml):
return json.dumps(pars) return json.dumps(pars)
def misskey_post(baseurl, content, channel=""):
print("Creating new post to", baseurl, ":", content)
req_url = baseurl + "/api/notes/create"
body = {
"visibility": "specified",
"text": content,
"localOnly": channel != "",
"i": "0v1t2poJeEh9RI74VqCCzJLHk9N6Wyds"
}
if channel != "":
body["channelId"] = channel
result = requests.post(req_url, json=body)
if result.status_code == 200:
return result
else:
print("Failed to post:", result.json()['error']['message'])
return False
def spider(rule_name, rss_url): def spider(rule_name, rss_url):
print("Fetch: ", rss_url) print("Fetch: [" + rule_name + "] ", rss_url)
time_start = time.time()
c = conn.cursor() c = conn.cursor()
result = xmltodict.parse(requests.get(rss_url).content) fetch = requests.get(rss_url)
if fetch.status_code != 200:
print("Failed to fetch")
return False
result = xmltodict.parse(fetch.content)
c.execute('INSERT INTO "main"."spider_log" ("rule_name", "rss_url", "result_json", "timestamp") ' c.execute('INSERT INTO "main"."spider_log" ("rule_name", "rss_url", "result_json", "timestamp") '
'VALUES (?, ?, ?, ?)', (rule_name, rss_url, json.dumps(result), time.time())) 'VALUES (?, ?, ?, ?)', (rule_name, rss_url, json.dumps(result), time.time()))
c.close() c.close()
time_end = time.time()
print("Fetch done in", time_end - time_start, "s")
return result return result
if __name__ == '__main__': if __name__ == '__main__':
print("Misskey X61 RSS Bot") print("Misskey X61 RSS Bot initialized")
spider("duowei", "https://rsshub.app/dwnews/yaowen/global") spider("duowei", "https://rsshub.app/dwnews/yaowen/global")
req = misskey_post("https://misskey.io", "Beep.. Beep Beep! Beep:"+str(time.time()))
conn.commit() conn.commit()
conn.close() conn.close()
time_end = time.time()
print("X61 bot: done in", time_end - time_start, "s")