Initial Bot Test, no advanced Features, only manual posting

This commit is contained in:
Neshura 2023-02-06 00:10:20 +01:00
parent 4737fc3a93
commit 1a6586343f
No known key found for this signature in database
GPG key ID: ACDF5B6EBECF6B0A
4 changed files with 182 additions and 0 deletions

2
.gitignore vendored Normal file
View file

@ -0,0 +1,2 @@
/venv/
/config.json

View file

@ -1 +1,2 @@
# ascendance-of-a-bookworm-bot
`python -m venv venv`

80
bot.py Normal file
View file

@ -0,0 +1,80 @@
from enum import Enum
import praw
import json
import requests
class PostFlair(str, Enum):
debug = "b9626520-a55b-11ed-8a89-de8567ca69ac"
light_novel = "4c6c8e7c-2f96-11ed-b4e8-8ad3482a4000"
manga = "48e0f284-2f96-11ed-8035-9e12ed32b10a"
anime = "4671b6fa-2f96-11ed-8f94-324864767feb"
pre_pub = "dc2829a2-a14e-11ed-962a-ee45985252e2"
class PostData():
title: str
link: str
image: str
flair: PostFlair
def __init__(self):
pass
def main():
data = loadConfig("config.json")
postConfigs = data['PostConfigs']
botConfig = data['BotConfig']
reddit = praw.Reddit(
client_id = botConfig['appId'],
client_secret = botConfig['appSecret'],
refresh_token = botConfig['refreshToken'],
redirect_uri = botConfig['redirectUri'],
user_agent = botConfig['userAgent']
)
sub = reddit.subreddit(botConfig['subReddit'])
#submissionList = reddit.redditor("MestionorasAvatar").submissions.new()
# Timer Logic will be implemented here
# Move into class to improve performance (less value copying)
if True:
feedUri = postConfigs[0]['feed']
postData = getFeedData(feedUri)
postData.flair = PostFlair[postConfigs[0]['flair']]
exists = True
#for submission in submissionList:
exists = False
if not exists:
createPost(sub, postData)
pass
def createPost(sub, postConfig):
sub.submit(title=postConfig.title, url=postConfig.link, flair_id=postConfig.flair.value, spoiler=True)
pass
def getFeedData(uri) -> PostData:
res = requests.get(uri)
newPost = PostData()
if res.ok:
data = res.json()
# return most recent entry
recentEntry = data['items'][0]
newPost.title = recentEntry['title']
newPost.link = recentEntry['url']
newPost.image = recentEntry['image']
return newPost
def loadConfig(filePath):
configFile = open(filePath)
data = json.load(configFile)
return data
if __name__ == "__main__":
main()

99
refreshtoken.py Normal file
View file

@ -0,0 +1,99 @@
#!/usr/bin/env python
"""This example demonstrates the flow for retrieving a refresh token.
In order for this example to work your application's redirect URI must be set to
http://localhost:8080.
This tool can be used to conveniently create refresh tokens for later use with your web
application OAuth2 credentials.
"""
import random
import socket
import sys
import praw
def receive_connection():
"""Wait for and then return a connected socket..
Opens a TCP connection on port 8080, and waits for a single client.
"""
server = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
server.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
server.bind(("localhost", 8080))
server.listen(1)
client = server.accept()[0]
server.close()
return client
def send_message(client, message):
"""Send message to client and close the connection."""
print(message)
client.send(f"HTTP/1.1 200 OK\r\n\r\n{message}".encode("utf-8"))
client.close()
def main():
"""Provide the program's entry point when directly executed."""
print(
"Go here while logged into the account you want to create a token for: "
"https://www.reddit.com/prefs/apps/"
)
print(
"Click the create an app button. Put something in the name field and select the"
" script radio button."
)
print("Put http://localhost:8080 in the redirect uri field and click create app")
client_id = input(
"Enter the client ID, it's the line just under Personal use script at the top: "
)
client_secret = input("Enter the client secret, it's the line next to secret: ")
commaScopes = input(
"Now enter a comma separated list of scopes, or all for all tokens: "
)
if commaScopes.lower() == "all":
scopes = ["*"]
else:
scopes = commaScopes.strip().split(",")
reddit = praw.Reddit(
client_id=client_id.strip(),
client_secret=client_secret.strip(),
redirect_uri="http://localhost:8080",
user_agent="praw_refresh_token_example",
)
state = str(random.randint(0, 65000))
url = reddit.auth.url(scopes, state, "permanent")
print(f"Now open this url in your browser: {url}")
sys.stdout.flush()
client = receive_connection()
data = client.recv(1024).decode("utf-8")
param_tokens = data.split(" ", 2)[1].split("?", 1)[1].split("&")
params = {
key: value for (key, value) in [token.split("=") for token in param_tokens]
}
if state != params["state"]:
send_message(
client,
f"State mismatch. Expected: {state} Received: {params['state']}",
)
return 1
elif "error" in params:
send_message(client, params["error"])
return 1
refresh_token = reddit.auth.authorize(params["code"])
send_message(client, f"Refresh token: {refresh_token}")
return 0
if __name__ == "__main__":
sys.exit(main())