Example #1
0
    def create_db():
        Config.load_config()
        host = Config.get_db_host_name()
        port = Config.get_db_port()
        db_name = Config.get_db_name()
        user = Config.get_db_user()
        password = Config.get_db_password()

        db = PostgresqlDatabase(db_name,
                                user=user,
                                password=password,
                                host=host,
                                port=port)
        return db
        r.raise_for_status()
        response = r.json()
        return response['token']

    def get_menu_entries(self):
        start_time = datetime.now()

        auth_token = self.get_token(os.environ['STI_API_USER'],
                                    os.environ['STI_API_PASSWORD'])
        header = {"Authorization": "Token " + auth_token}
        # TODO: remove verify=False if API has proper ssl certificate
        r = requests.get(self.URL + "menuentries/",
                         headers=header,
                         verify=False)

        self.measure_api_call("get_menu_entries", start_time)
        r.raise_for_status()
        response = r.json()
        return response

    @staticmethod
    def measure_api_call(method_name, start_time):
        time_elapsed = datetime.now() - start_time
        print(method_name +
              ' API-Request took (hh:mm:ss.ms) {}'.format(time_elapsed))


if __name__ == '__main__':
    Config.load_config()
    print(STIMenuFetcher().get_menu_entries())