Initial Bot Test, no advanced Features, only manual posting
This commit is contained in:
parent
4737fc3a93
commit
1a6586343f
4 changed files with 182 additions and 0 deletions
2
.gitignore
vendored
Normal file
2
.gitignore
vendored
Normal file
|
@ -0,0 +1,2 @@
|
|||
/venv/
|
||||
/config.json
|
|
@ -1 +1,2 @@
|
|||
# ascendance-of-a-bookworm-bot
|
||||
`python -m venv venv`
|
80
bot.py
Normal file
80
bot.py
Normal 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
99
refreshtoken.py
Normal 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())
|
Loading…
Reference in a new issue