Skip to content
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

Pages and sidebar #15

Open
wants to merge 4 commits into
base: main
Choose a base branch
from
Open

Pages and sidebar #15

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
4 commits
Select commit Hold shift + click to select a range
b2e85ce
updated actions endpoint
OmriGez Aug 15, 2024
de1e7f1
added additional data handling
OmriGez Sep 22, 2024
c4057da
reverted US API
OmriGez Sep 22, 2024
c75e7c2
Merge branch 'main' into pages_and_sidebar
OmriGez Sep 22, 2024
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
72 changes: 64 additions & 8 deletions clean.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -16,20 +16,76 @@

systemBlueprints = ["_user", "_team"]

print("Deleting actions")
res = requests.get(f'{API_URL}/actions?version=v2', headers=headers)
actions = res.json()["actions"]
for action in actions:
print(f"deleting action {action['identifier']}")
res = requests.delete(f'{API_URL}/actions/{action["identifier"]}', headers=headers)
if res.ok != True:
print("error while deleting action: "+ json.dumps(res.json()))

print("Deleting pages")
res = requests.get(f'{API_URL}/pages', headers=headers)
pages = res.json()["pages"]
for page in pages:
print(f"deleting page {page['identifier']}")
res = requests.delete(f'{API_URL}/pages/{page["identifier"]}', headers=headers)
if res.ok != True:
print("error while deleting page: "+ json.dumps(res.json()))

print("Getting blueprints")
res = requests.get(f'{API_URL}/blueprints', headers=headers)
resp = res.json()["blueprints"]
blueprints = res.json()["blueprints"]
failed_blueprints = []

for blueprint in resp:
for blueprint in blueprints:
if blueprint["identifier"] not in systemBlueprints:
print(f"deleting entities of blueprint {blueprint['identifier']}")
print(f"Deleting entities of blueprint {blueprint['identifier']}")
res = requests.delete(f'{API_URL}/blueprints/{blueprint["identifier"]}/all-entities', headers=headers)
if res.ok != True:
print("error while deleting entities: "+ json.dumps(res.json()))
print(f"deleting blueprint {blueprint['identifier']}")
if not res.ok:
print(f"Error while deleting entities: {json.dumps(res.json())}")
failed_blueprints.append(blueprint["identifier"])
continue

print(f"Deleting blueprint {blueprint['identifier']}")
res = requests.delete(f'{API_URL}/blueprints/{blueprint["identifier"]}', headers=headers)
if res.ok != True:
print("error while deleting blueprint: "+ json.dumps(res.json()))
if not res.ok:
print(f"Error while deleting blueprint: {json.dumps(res.json())}")
failed_blueprints.append(blueprint["identifier"])

return failed_blueprints

for attempt in range(5):
if not failed_blueprints:
break

print(f"Retrying deletion... Attempt {attempt + 1}")
remaining_failed = []

for identifier in failed_blueprints:
print(f"Retrying deletion of blueprint {identifier}")
res = requests.delete(f'{API_URL}/blueprints/{identifier}/all-entities', headers=headers)
if not res.ok:
print(f"Error while deleting entities: {json.dumps(res.json())}")
remaining_failed.append(identifier)
continue

res = requests.delete(f'{API_URL}/blueprints/{identifier}', headers=headers)
if not res.ok:
print(f"Error while deleting blueprint: {json.dumps(res.json())}")
remaining_failed.append(identifier)

failed_blueprints = remaining_failed
if failed_blueprints:
print(f"Some deletions still failed. Retrying in 5 seconds...")
time.sleep(5)

if failed_blueprints:
print(f"Failed to delete the following blueprints after {retries} attempts: {failed_blueprints}")
else:
print("All blueprints deleted successfully.")


print("Getting teams")
res = requests.get(f'{API_URL}/teams', headers=headers)
Expand Down
67 changes: 59 additions & 8 deletions main.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -7,6 +7,7 @@
import openpyxl

API_URL = 'https://api.getport.io/v1'
US_API_URL = 'https://api.us.getport.io/v1'

global error
error = False
Expand Down Expand Up @@ -93,13 +94,22 @@ def getScorecards():
resp = res.json()["scorecards"]
return resp

def getSidebar():
res = requests.get(f'{API_URL}/sidebars/catalog', headers=old_headers)
sidebar = res.json()['sidebar']['items']

def getActions():
print("Getting actions")
res = requests.get(f'{API_URL}/actions', headers=old_headers)
res = requests.get(f'{API_URL}/actions?version=v2', headers=old_headers)
resp = res.json()["actions"]
return resp

def getPages():
print("Getting pages")
res = requests.get(f'{API_URL}/pages', headers=old_headers)
resp = res.json()["pages"]
return resp

def getTeams():
print("Getting teams")
res = requests.get(f'{API_URL}/teams', headers=old_headers)
Expand All @@ -124,7 +134,7 @@ def postBlueprints(blueprints):
bp["aggregationProperties"] = {}
bp["relations"] = {}
bp["mirrorProperties"] = {}
res = requests.post(f'{API_URL}/blueprints', headers=new_headers, json=bp)
res = requests.post(f'{API_URL}/blueprints?create_catalog_page=false', headers=new_headers, json=bp)
if res.ok != True:
print("error posting blueprint:" + json.dumps(res.json()))
error = True
Expand All @@ -145,6 +155,31 @@ def postBlueprints(blueprints):
print("error patching blueprint:" + json.dumps(res.json()))
error = True

def postPages(pages, sidebar):
for catalog_item in sidebar:
if catalog_item['sidebarType'] == 'page': #catalog_item is a page
pageToPost = next((page for page in pages if page['identifier'] == catalog_item['identifier']), None)
print(f"posting page {catalog_item['identifier']}")
res = requests.post(f'{API_URL}/pages', headers=new_headers, json=pageToPost)
if res.ok != True:
print("error posting page:" + json.dumps(res.json()))
error = True
else: #catalog_item is a folder
catalog_item.pop("createdAt", None)
catalog_item.pop("updatedAt", None)
catalog_item.pop("createdBy", None)
catalog_item.pop("updatedBy", None)
catalog_item.pop("_orgId", None)
catalog_item.pop("_id", None)
catalog_item.pop("sidebar", None)
catalog_item.pop("sidebarType", None)
res = requests.post(f'{API_URL}/sidebars/catalog/folders', headers=new_headers, json=catalog_item)
if res.ok != True:
print("error posting catalog item:" + json.dumps(res.json()))
error = True



def postExcelEntities(entities, blueprints):
for item in entities.values():
for entity in item:
Expand Down Expand Up @@ -215,7 +250,6 @@ def postActions(actions):
print("Posting actions")
for action in actions:
print(f"posting action {action['identifier']}")
action.pop("id", None)
blueprint = action.pop("blueprint", None)
if pd.isna(action.get("description", "")): # check if description is NaN
action["description"] = "" # set description to empty string
Expand Down Expand Up @@ -265,6 +299,8 @@ def main():
blueprints = getBlueprints()
scorecards = getScorecards()
actions = getActions()
sidebar = getSidebar()
pages = getPages()
teams = getTeams()
entities = {}
for blueprint in blueprints:
Expand All @@ -280,18 +316,26 @@ def main():
json.dump(actions, outfile)
with open('bk-teams.json', 'w') as outfile:
json.dump(teams, outfile)
with open('bk-pages.json', 'w') as outfile:
json.dump(pages, outfile)
with open('bk-sidebar', 'w') as outfile:
json.dump(sidebar, outfile)
with open('bk-entities.json', 'w') as outfile:
json.dump(entities, outfile)
else:
df_blueprints = pd.DataFrame(blueprints).map(lambda x: json.dumps(x) if isinstance(x, dict) or isinstance(x,list) else x)
df_scorecards = pd.DataFrame(scorecards).map(lambda x: json.dumps(x) if isinstance(x, dict) or isinstance(x,list) else x)
df_actions = pd.DataFrame(actions).map(lambda x: json.dumps(x) if isinstance(x, dict) or isinstance(x,list) else x)
df_teams = pd.DataFrame(teams).map(lambda x: json.dumps(x) if isinstance(x, dict) or isinstance(x,list) else x)
df_pages = pd.DataFrame(pages).map(lambda x: json.dumps(x) if isinstance(x, dict) or isinstance(x,list) else x)
df_sidebar = pd.DataFrame(sidebar).map(lambda x: json.dumps(x) if isinstance(x, dict) or isinstance(x,list) else x)
with pd.ExcelWriter('bk-data.xlsx') as writer:
df_blueprints.to_excel(writer, sheet_name='Blueprints', index=False)
df_scorecards.to_excel(writer, sheet_name='Scorecards', index=False)
df_actions.to_excel(writer, sheet_name='Actions', index=False)
df_teams.to_excel(writer, sheet_name='Teams', index=False)
df_pages.to_excel(writer, sheet_name='Pages', index=False)
df_sidebar.to_excel(writer, sheet_name='Sidebar', index=False)
for blueprint, entity_list in entities.items():
for entity in entity_list:
for key in entity["properties"]:
Expand All @@ -303,7 +347,7 @@ def main():
entity.pop("relations", None)
df = pd.DataFrame(entity_list).map(lambda x: json.dumps(x) if isinstance(x, dict) or isinstance(x,list) else x)
df.to_excel(writer, sheet_name=blueprint, index=False)

if RUN_MODE == "migrate" or RUN_MODE == "restore":
if RUN_MODE == "restore":
if FORMAT == "tar":
Expand All @@ -315,17 +359,23 @@ def main():
actions = json.load(json_file)
with open('bk-teams.json') as json_file:
teams = json.load(json_file)
with open('bk-pages') as json_file:
pages = json.load(json_file)
with open('bk-sidebar') as json_file:
sidebar = json.load(json_file)
with open('bk-entities.json') as json_file:
entities = json.load(json_file)
else:
blueprints = parser(pd.read_excel(os.getenv("FILE_NAME"), sheet_name='Blueprints').to_dict(orient='records'))
scorecards = parser((pd.read_excel(os.getenv("FILE_NAME"), sheet_name='Scorecards').to_dict(orient='records')))
actions = parser((pd.read_excel(os.getenv("FILE_NAME"), sheet_name='Actions').to_dict(orient='records')))
teams = parser((pd.read_excel(os.getenv("FILE_NAME"), sheet_name='Teams').to_dict(orient='records')))
pages = parser((pd.read_excel(os.getenv("FILE_NAME"), sheet_name='Pages').to_dict(orient='records')))
sidebar = parser((pd.read_excel(os.getenv("FILE_NAME"), sheet_name='Sidebar').to_dict(orient='records')))
entities = {}
sheet_names = pd.ExcelFile(os.getenv("FILE_NAME")).sheet_names
for sheet in sheet_names:
if sheet not in ['Blueprints', 'Scorecards', 'Actions', 'Teams']: # Skip non-entity sheets
if sheet not in ['Blueprints', 'Scorecards', 'Actions', 'Teams', 'Pages', 'Sidebar']: # Skip non-entity sheets
df = parser(pd.read_excel(os.getenv("FILE_NAME"), sheet_name=sheet).to_dict(orient='records'))
for entity in df:
entity["properties"] = {}
Expand All @@ -344,11 +394,12 @@ def main():
postBlueprints(blueprints)
postScorecards(scorecards)
postActions(actions)
postPages(pages, sidebar)
postTeams(teams)
if FORMAT == "tar":
postEntities(entities, blueprints)
postEntities(entities, blueprints)
else:
postExcelEntities(entities, blueprints)
postExcelEntities(entities, blueprints)
if error:
print("Errors occured during migration, please check logs")
elif teamError:
Expand All @@ -357,4 +408,4 @@ def main():
print("No errors were caught during migration")

if __name__ == "__main__":
main()
main()