67 lines
2.4 KiB
Python
67 lines
2.4 KiB
Python
"""
|
|
This is a one-time fix script where we need to fix some content of the accounts created.
|
|
"""
|
|
|
|
# Imports:
|
|
import os
|
|
import pandas as pd
|
|
from helpers import tcaoff
|
|
from utils_v2.string import json
|
|
from utils_v2.system import files
|
|
|
|
# File paths:
|
|
file_dir = files.get_file_directory(include_filename = False)
|
|
proj_dir = files.get_parent_directory(file_dir, depth = 1)
|
|
tcaoff_creds_file = os.path.join(proj_dir, "creds", "tcaoff.json")
|
|
|
|
# Load the creds:
|
|
tcaoff_creds = json.from_file(tcaoff_creds_file)
|
|
|
|
# Log in:
|
|
tcaoff.login(tcaoff_creds)
|
|
|
|
# Load the team list:
|
|
team_list = tcaoff.team_list(tcaoff_creds)
|
|
|
|
# Loop through the list,
|
|
# check if `applicantNotes` in `json_notes` is a string,
|
|
# if it is a string, convert it to a dict/list and update the user:
|
|
for team in team_list:
|
|
|
|
# Extract info:
|
|
json_notes = json.from_string(team.get("json_notes", {}))
|
|
app_notes = json_notes.get("applicantNotes")
|
|
if isinstance(app_notes, str):
|
|
app_notes = json.from_string(app_notes)
|
|
if not app_notes: continue
|
|
|
|
# Fix the app notes:
|
|
cosec_notes = app_notes.get("cosec", {})
|
|
cosec_notes = {
|
|
"User ID": cosec_notes.get("User ID", cosec_notes.get("UserID")),
|
|
"User Name": cosec_notes.get("User Name", cosec_notes.get("UserName")),
|
|
"Category Name": cosec_notes.get("Category Name", cosec_notes.get("CategoryName")),
|
|
"Grade Name": cosec_notes.get("Grade Name", cosec_notes.get("GradeName")),
|
|
"Branch Name": cosec_notes.get("Branch Name", cosec_notes.get("BranchName")),
|
|
"Department Name": cosec_notes.get("Department Name", cosec_notes.get("DepartmentName")),
|
|
"Direct Reporting": cosec_notes.get("Direct Reporting", cosec_notes.get("DirectReporting")),
|
|
"Direct Reporting ID": cosec_notes.get("Direct Reporting ID", cosec_notes.get("DirectReportingID")),
|
|
"Level-1 ID": cosec_notes.get("Level-1 ID", cosec_notes.get("Level-1ID")),
|
|
}
|
|
print("COSEC NOTES:", cosec_notes)
|
|
|
|
# Save the JSON Notes again:
|
|
app_notes["cosec"] = cosec_notes
|
|
json_notes["applicantNotes"] = app_notes
|
|
team["json_notes"] = json.to_string(json_notes, no_space = True)
|
|
|
|
# Make a pandas DataFrame:
|
|
team_df = pd.DataFrame(team_list)
|
|
|
|
# Log out:
|
|
tcaoff.logout(tcaoff_creds)
|
|
|
|
# Show and save the DF:
|
|
print(team_df[:10].to_string())
|
|
team_df[["user_id", "json_notes"]].to_csv(r"C:\Users\Khushal P Soonderji\Downloads\20260122_cosec_team_fix.csv", index = False)
|