109 lines
4.1 KiB
Python
109 lines
4.1 KiB
Python
|
import praw
|
||
|
|
import os.path
|
||
|
|
import json
|
||
|
|
import time
|
||
|
|
|
||
|
|
# Create default config/config.py if it doesn't exist and exit to prompt manual editing
|
||
|
|
import sys
|
||
|
|
|
||
|
|
default_config_path = os.path.join('config', 'config.py')
|
||
|
|
def env_or_default(var, default):
|
||
|
|
return os.environ.get(var, default)
|
||
|
|
|
||
|
|
def write_config_from_env():
|
||
|
|
os.makedirs('config', exist_ok=True)
|
||
|
|
with open(default_config_path, 'w') as f:
|
||
|
|
f.write(
|
||
|
|
f'username = "{env_or_default("USERNAME", "")}"\n'
|
||
|
|
f'password = "{env_or_default("PASSWORD", "")}"\n'
|
||
|
|
f'client_id = "{env_or_default("CLIENT_ID", "")}"\n'
|
||
|
|
f'client_secret = "{env_or_default("CLIENT_SECRET", "")}"\n'
|
||
|
|
f'user_agent = "{env_or_default("USER_AGENT", "Flair Timer Comment Bot" )}"\n'
|
||
|
|
'\n'
|
||
|
|
f'subreddit = "{env_or_default("SUBREDDIT", "")}"\n'
|
||
|
|
f'flair_text = "{env_or_default("FLAIR_TEXT", "Waiting for OP")}"\n'
|
||
|
|
f'interval = {env_or_default("INTERVAL", "30")}\n'
|
||
|
|
f'hours = {env_or_default("HOURS", "48")}\n'
|
||
|
|
f'messagetitle = "{env_or_default("MESSAGETITLE", "Modmail Notification")}"\n'
|
||
|
|
f'searchlimit = {env_or_default("SEARCHLIMIT", "600")}\n'
|
||
|
|
)
|
||
|
|
print(f"Configuration file auto-populated from environment variables at {default_config_path}.")
|
||
|
|
|
||
|
|
# Check if config file is missing or empty
|
||
|
|
populate_config = False
|
||
|
|
if not os.path.exists(default_config_path):
|
||
|
|
populate_config = True
|
||
|
|
else:
|
||
|
|
try:
|
||
|
|
with open(default_config_path, 'r') as f:
|
||
|
|
content = f.read().strip()
|
||
|
|
if not content:
|
||
|
|
populate_config = True
|
||
|
|
except Exception:
|
||
|
|
populate_config = True
|
||
|
|
|
||
|
|
if populate_config:
|
||
|
|
write_config_from_env()
|
||
|
|
sys.exit(0)
|
||
|
|
|
||
|
|
if not os.path.exists(default_config_path):
|
||
|
|
sys.exit(0)
|
||
|
|
import config
|
||
|
|
|
||
|
|
def authentication():
|
||
|
|
print ("Authenticating...")
|
||
|
|
reddit = praw.Reddit(username = config.username,
|
||
|
|
password = config.password,
|
||
|
|
client_id = config.client_id,
|
||
|
|
client_secret = config.client_secret,
|
||
|
|
user_agent = config.user_agent)
|
||
|
|
print ("Authenticated as {}.".format(reddit.user.me()))
|
||
|
|
return reddit
|
||
|
|
|
||
|
|
def main(reddit, posts: dict):
|
||
|
|
while True:
|
||
|
|
for submission in reddit.subreddit(config.subreddit).new(limit=config.searchlimit):
|
||
|
|
if not submission.saved:
|
||
|
|
if submission.id not in posts.keys() and submission.link_flair_text == config.flair_text:
|
||
|
|
posts[submission.id] = time.time()
|
||
|
|
print(f"Post {submission} has been flaired {config.flair_text}")
|
||
|
|
if submission.id in posts.keys() and submission.link_flair_text != config.flair_text:
|
||
|
|
posts.pop(submission.id)
|
||
|
|
print(f"Post {submission} has been unflaired {config.flair_text}")
|
||
|
|
|
||
|
|
for submission in posts:
|
||
|
|
if time.time() > posts[submission] + (config.hours * 60 * 60):
|
||
|
|
posts.pop(submission)
|
||
|
|
reddit.submission(submission).save()
|
||
|
|
data = {
|
||
|
|
"subject": config.messagetitle,
|
||
|
|
"text": f"It has been {config.hours/24} day/s since this was flaired [{config.flair_text}](https://old.reddit.com{reddit.submission(submission).permalink})",
|
||
|
|
"to": "/r/{}".format(config.subreddit),
|
||
|
|
}
|
||
|
|
reddit.post("api/compose/", data=data)
|
||
|
|
print(f"Post {submission} has been flaired {config.flair_text} for {config.hours * 60} minutes, sent modmail")
|
||
|
|
break
|
||
|
|
|
||
|
|
save_posts(posts)
|
||
|
|
time.sleep(config.interval)
|
||
|
|
|
||
|
|
def load_posts():
|
||
|
|
if not os.path.exists("config/posts.json"):
|
||
|
|
with open("config/posts.json", "w+") as file:
|
||
|
|
json.dump({}, file)
|
||
|
|
with open("config/posts.json", "r+") as file:
|
||
|
|
data = json.load(file)
|
||
|
|
return data
|
||
|
|
|
||
|
|
def save_posts(data):
|
||
|
|
with open('config/posts.json', 'w+') as file:
|
||
|
|
json.dump(data, file)
|
||
|
|
|
||
|
|
|
||
|
|
while True:
|
||
|
|
try:
|
||
|
|
posts = load_posts()
|
||
|
|
main(reddit = authentication(), posts = posts)
|
||
|
|
except Exception as e:
|
||
|
|
print(e)
|