Files
arXiv_abstract_bot/botlib.py
T
2020-05-06 11:19:22 +08:00

39 lines
925 B
Python

import praw
import logging
import requests
import bs4
import html2text
import time, os
import shelve
import re
from prawcore import NotFound
import logging
import sys
logging.basicConfig(stream=sys.stdout, level=logging.INFO)
def get_logger():
logger = logging.getLogger(__name__)
logger.setLevel(logging.DEBUG)
return logger
def get_memcache_client(name):
client = shelve.open(name)
return client
def get_bot():
PRAW_CLIENT_ID = os.environ.get('PRAW_CLIENT_ID')
PRAW_CLIENT_SECRET = os.environ.get('PRAW_CLIENT_SECRET')
PRAW_PASSWORD = os.environ.get('PRAW_PASSWORD')
PRAW_USERNAME = os.environ.get('PRAW_USERNAME')
PRAW_USERAGENT = os.environ.get('PRAW_USERAGENT')
return praw.Reddit(
username=PRAW_USERNAME,
password=PRAW_PASSWORD,
client_id=PRAW_CLIENT_ID,
client_secret=PRAW_CLIENT_SECRET,
user_agent=PRAW_USERAGENT
)