beepboop/API.py at main · bearstonem/beepboop · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
import configparser
import sys


class API:

    def __init__(self):
        self.config = configparser.ConfigParser()
        self.config.read("config")

        self.API_KEY = self.config.get("SECRETS", "ApiKey", fallback="NoKey")
        self.API_SECRET = self.config.get("SECRETS", "ApiSecret", fallback="NoSecret")
        self.ACCESS_TOKEN = self.config.get("SECRETS", "AccessToken", fallback="NoAccessToken")
        self.ACCESS_TOKEN_SECRET = self.config.get("SECRETS", "AccessTokenSecret", fallback="NoAccessTokenSecret")

        if "NoKey" in self.API_KEY:
            print("No API key found. Please set one in the config file.")
            sys.exit(0)

        if "NoSecret" in self.API_SECRET:
            print("No API secret found. Please set one in the config file.")
            sys.exit(0)

        if "NoAccessToken" in self.ACCESS_TOKEN:
            print("No API bearer found. Please set one in the config file.")
            sys.exit(0)

        if "NoAccessTokenSecret" in self.ACCESS_TOKEN_SECRET:
            print("No API bearer found. Please set one in the config file.")
            sys.exit(0)

    def get_api_key(self):
        return self.API_KEY

    def get_api_secret(self):
        return self.API_SECRET

    def get_access_token(self):
        return self.ACCESS_TOKEN

    def get_access_token_secret(self):
        return self.ACCESS_TOKEN_SECRET