refactor: encaps create/update logic in upsert
This commit is contained in:
parent
fa67103b20
commit
3940cbdbde
51
agent.py
51
agent.py
|
@ -244,6 +244,36 @@ class ServerManager:
|
|||
return server['id']
|
||||
return None
|
||||
|
||||
def upsert_server(self, post_data):
|
||||
server_id = self.existing_server_id(post_data)
|
||||
|
||||
if server_id:
|
||||
logging.info('Server already exists with id: {}, Updating...'.format(server_id))
|
||||
response = self.update_server(post_data, server_id)
|
||||
else:
|
||||
logging.info('Server does not exist, Creating...')
|
||||
response = self.create_server(post_data)
|
||||
|
||||
# Extract the server_id from the response
|
||||
server_id = response.get('server_id', None)
|
||||
if server_id is None:
|
||||
logging.error('Failed to get server_id from response: {}'.format(response))
|
||||
raise ValueError('Failed to get server_id from response')
|
||||
|
||||
return server_id
|
||||
|
||||
def upsert_note(self, note_data, server_id):
|
||||
note_data['service_id'] = server_id
|
||||
try:
|
||||
note = self.get_note(server_id)
|
||||
except urllib.error.HTTPError:
|
||||
note = None
|
||||
|
||||
if note:
|
||||
return self.update_note(note_data, server_id)
|
||||
else:
|
||||
return self.create_note(note_data)
|
||||
|
||||
def validate_env_vars():
|
||||
api_key = os.getenv('AGENT_API')
|
||||
host = os.getenv('HOST')
|
||||
|
@ -263,26 +293,11 @@ def main():
|
|||
|
||||
server_manager = ServerManager(host, api_key)
|
||||
|
||||
# Check if the server already exists
|
||||
server_id = server_manager.existing_server_id(post_data)
|
||||
|
||||
# If the server exists, update it
|
||||
if server_id:
|
||||
logging.info('Server already exists with id: {}, Updating...'.format(server_id))
|
||||
logging.info(server_manager.update_server(post_data, server_id))
|
||||
else:
|
||||
logging.info('Server does not exist, Creating...')
|
||||
logging.info(server_manager.create_server(post_data))
|
||||
server_id = server_manager.upsert_server(post_data)
|
||||
logging.info('Server id: {}'.format(server_id))
|
||||
|
||||
note_data = server_data.create_note_data(server_id)
|
||||
try:
|
||||
note = server_manager.get_note(server_id)
|
||||
except urllib.error.HTTPError:
|
||||
note = None
|
||||
if note:
|
||||
server_manager.update_note(note_data, server_id)
|
||||
else:
|
||||
server_manager.create_note(note_data)
|
||||
server_manager.upsert_note(note_data, server_id)
|
||||
|
||||
if __name__ == '__main__':
|
||||
main()
|
||||
|
|
Loading…
Reference in New Issue
Block a user