Skip to content

Refactor to make more import friendly #42

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
chm0dx wants to merge 1 commit into
base: master
Choose a base branch
from
Open

Refactor to make more import friendly #42

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
104 changes: 56 additions & 48 deletions fire.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -13,20 +13,20 @@
import configparser
from typing import Tuple, Callable

class FireProxException(Exception):
pass

class FireProx(object):
def __init__(self, arguments: argparse.Namespace, help_text: str):
self.profile_name = arguments.profile_name
self.access_key = arguments.access_key
self.secret_access_key = arguments.secret_access_key
self.session_token = arguments.session_token
self.region = arguments.region
self.command = arguments.command
self.api_id = arguments.api_id
self.url = arguments.url
def __init__(self, **kwargs):
self.profile_name = None
self.command = None
self.access_key = None
self.secret_access_key = None
self.session_token = None
self.region = None
self.api_list = []
self.client = None
self.help = help_text
self.__dict__.update(kwargs)

if self.access_key and self.secret_access_key:
if not self.region:
Expand All @@ -35,9 +35,6 @@ def __init__(self, arguments: argparse.Namespace, help_text: str):
if not self.load_creds():
self.error('Unable to load AWS credentials')

if not self.command:
self.error('Please provide a valid command')

def __str__(self):
return 'FireProx()'

Expand Down Expand Up @@ -77,7 +74,7 @@ def load_creds(self) -> bool:
config_profile_section = f'profile {self.profile_name}'
if self.profile_name in credentials:
if config_profile_section not in config:
print(f'Please create a section for {self.profile_name} in your ~/.aws/config file')
self.error(f'Please create a section for {self.profile_name} in your ~/.aws/config file')
return False
self.region = config[config_profile_section].get('region', 'us-east-1')
try:
Expand Down Expand Up @@ -123,11 +120,9 @@ def load_creds(self) -> bool:
return False

def error(self, error):
print(self.help)
sys.exit(error)
raise FireProxException(error)

def get_template(self):
url = self.url
def get_template(self, url):
if url[-1] == '/':
url = url[:-1]

Expand Down Expand Up @@ -239,17 +234,15 @@ def create_api(self, url):
if not url:
self.error('Please provide a valid URL end-point')

print(f'Creating => {url}...')

template = self.get_template()
template = self.get_template(url)
response = self.client.import_rest_api(
parameters={
'endpointConfigurationTypes': 'REGIONAL'
},
body=template
)
resource_id, proxy_url = self.create_deployment(response['id'])
self.store_api(
return self.store_api(
response['id'],
response['name'],
response['createdDate'],
Expand All @@ -268,7 +261,6 @@ def update_api(self, api_id, url):

resource_id = self.get_resource(api_id)
if resource_id:
print(f'Found resource {resource_id} for {api_id}!')
response = self.client.update_integration(
restApiId=api_id,
resourceId=resource_id,
Expand Down Expand Up @@ -299,6 +291,7 @@ def delete_api(self, api_id):
return False

def list_api(self, deleted_api_id=None):
results = []
response = self.client.get_rest_apis()
for item in response['items']:
try:
Expand All @@ -308,15 +301,18 @@ def list_api(self, deleted_api_id=None):
proxy_url = self.get_integration(api_id).replace('{proxy}', '')
url = f'https://{api_id}.execute-api.{self.region}.amazonaws.com/fireprox/'
if not api_id == deleted_api_id:
print(f'[{created_dt}] ({api_id}) {name}: {url} => {proxy_url}')
results.append(f'[{created_dt}] ({api_id}) {name}: {url} => {proxy_url}')
except:
pass

return response['items']
if deleted_api_id:
return response['items']
else:
return results

def store_api(self, api_id, name, created_dt, version_dt, url,
resource_id, proxy_url):
print(
return(
f'[{created_dt}] ({api_id}) {name} => {proxy_url} ({url})'
)

Expand Down Expand Up @@ -389,29 +385,41 @@ def main():
:return:
"""
args, help_text = parse_arguments()
fp = FireProx(args, help_text)
if args.command == 'list':
print(f'Listing API\'s...')
result = fp.list_api()

elif args.command == 'create':
result = fp.create_api(fp.url)

elif args.command == 'delete':
result = fp.delete_api(fp.api_id)
success = 'Success!' if result else 'Failed!'
print(f'Deleting {fp.api_id} => {success}')

elif args.command == 'update':
print(f'Updating {fp.api_id} => {fp.url}...')
result = fp.update_api(fp.api_id, fp.url)
success = 'Success!' if result else 'Failed!'
print(f'API Update Complete: {success}')

else:
print(f'[ERROR] Unsupported command: {args.command}\n')

try:
if not args.command:
raise FireProxException('Please provide a valid command')

fp = FireProx(**vars(args))
if args.command == 'list':
print(f'Listing API\'s...')
results = fp.list_api()
for result in results:
print(result)

elif args.command == 'create':
if not args.url:
raise FireProxException('Please provide a valid URL end-point')
print(f'Creating => {args.url}...')
result = fp.create_api(fp.url)
print(result)

elif args.command == 'delete':
result = fp.delete_api(fp.api_id)
success = 'Success!' if result else 'Failed!'
print(f'Deleting {fp.api_id} => {success}')

elif args.command == 'update':
print(f'Updating {fp.api_id} => {fp.url}...')
result = fp.update_api(fp.api_id, fp.url)
success = 'Success!' if result else 'Failed!'
print(f'API Update Complete: {success}')

else:
raise FireProxException(f'[ERROR] Unsupported command: {args.command}')
except FireProxException as ex:
print(help_text)
sys.exit(1)
sys.exit(ex)


if __name__ == '__main__':
Expand Down