52 líneas
1.8 KiB
Python
52 líneas
1.8 KiB
Python
import getpass
|
|
import os
|
|
import sys
|
|
|
|
# Returns the parameter from the specified file
|
|
def get_parameter( parameter, file_path ):
|
|
# Check if secrets file exists
|
|
if not os.path.isfile(file_path):
|
|
print("File %s not found, asking."%file_path)
|
|
write_parameter( parameter, file_path )
|
|
#sys.exit(0)
|
|
|
|
# Find parameter in file
|
|
with open( file_path ) as f:
|
|
for line in f:
|
|
if line.startswith( parameter ):
|
|
return line.replace(parameter + ":", "").strip()
|
|
|
|
# Cannot find parameter, exit
|
|
print(file_path + " Missing parameter %s "%parameter)
|
|
sys.exit(0)
|
|
|
|
def write_parameter( parameter, file_path ):
|
|
if not os.path.exists('config'):
|
|
os.makedirs('config')
|
|
print("Setting up Twitter keys & tokens ...")
|
|
print("\n")
|
|
api_key = input("API Key: ")
|
|
api_key_secret = input("API Key Secret: ")
|
|
access_token = input("Access Token: ")
|
|
access_token_secret = input("Access Token Secret: ")
|
|
|
|
with open(file_path, "w") as text_file:
|
|
print("api_key: {}".format(api_key), file=text_file)
|
|
print("api_key_secret: {}".format(api_key_secret), file=text_file)
|
|
print("access_token: {}".format(access_token), file=text_file)
|
|
print("access_token_secret: {}".format(access_token_secret), file=text_file)
|
|
|
|
print('\nDone!')
|
|
|
|
###############################################################################
|
|
# main
|
|
|
|
if __name__ == '__main__':
|
|
|
|
# Load configuration from config file
|
|
twitter_config_filepath = "config/keys_config.txt"
|
|
api_key = get_parameter("api_key", twitter_config_filepath)
|
|
api_key_secret = get_parameter("api_key_secret", twitter_config_filepath)
|
|
access_token = get_parameter("access_token", twitter_config_filepath)
|
|
access_token_secret = get_parameter("access_token_secret", twitter_config_filepath)
|