diff --git a/main.py b/main.py index fadba3b..fc96a7f 100644 --- a/main.py +++ b/main.py @@ -10,21 +10,27 @@ import configparser import sqlite3 import logging -import instagrapi import datetime as dt import threading import random import time +import os +import environ + def load_settings() -> None: global settings - if not os.path.exists('bot_settings.ini'): - sys.exit('FAILED TO FIND SETTING FILE!') + env = environ.Env() + env.read_env() + if not os.path.exists('.env'): + sys.exit('FAILED TO FIND ENV FILE!') else: - config = configparser.ConfigParser() - config.read('bot_settings.ini') - settings = config['BOT'] + settings = { + 'client_id': env('INSTAGRAM_CLIENT_ID'), + 'client_secret': env('INSTAGRAM_CLIENT_SECRET'), + 'owner_id': env('OWNER_ID') + } def initiate_db() -> None: