1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
|
import os
import sys
from dotenv import load_dotenv
from selenium import webdriver
from selenium.webdriver.chrome.service import Service as ChromeService
from selenium.webdriver.common.action_chains import ActionChains
from selenium.webdriver.common.by import By
from selenium.webdriver.support import expected_conditions as EC
from selenium.webdriver.support.ui import WebDriverWait
from webdriver_manager.chrome import ChromeDriverManager
# Loading the .env file
load_dotenv()
GITHUB_USERNAME = os.environ.get('GITHUB_USERNAME', '')
GITHUB_PASSWORD = os.environ.get('GITHUB_PASSWORD', '')
SKIP_COUNT = int(sys.argv[1] if len(sys.argv) > 1 else '0')
FETCH_COUNT = int(sys.argv[2] if len(sys.argv) > 2 else '100')
sys.tracebacklimit = 0
# Set up the WebDriver
service = ChromeService(ChromeDriverManager().install())
driver = webdriver.Chrome(service=service)
wait = WebDriverWait(driver, 10)
def login_with_github():
driver.get('http://localhost/#/login') # Replace with your actual login page URL
try:
print("Waiting for the GitHub login button to be clickable...")
github_button = wait.until(
EC.element_to_be_clickable((By.XPATH, '//*[@id="root"]/main/div/button[2]/span[1]'))
)
print("GitHub login button is clickable now.")
github_button.click()
print("Clicked the GitHub login button.")
print("Waiting for GitHub login page to load...")
wait.until(
EC.presence_of_element_located((By.ID, 'login_field'))
)
print("GitHub login page loaded.")
username_field = driver.find_element(By.ID, 'login_field')
username_field.send_keys(GITHUB_USERNAME)
password_field = driver.find_element(By.ID, 'password')
password_field.send_keys(GITHUB_PASSWORD)
sign_in_button = driver.find_element(By.NAME, 'commit')
sign_in_button.click()
print("Submitted GitHub login form.")
except Exception as e:
print(f"An error occurred during GitHub login: {str(e)}")
sys.exit(1)
# Wait for the Send SMS page
try:
print("Waiting for the Send SMS button to be clickable...")
send_sms_button = wait.until(
EC.presence_of_element_located((By.XPATH, '//*[@id="login"]/div[2]/div[2]/form/button'))
)
print("Send SMS button is clickable now.")
send_sms_button.click()
print("Clicked the Send SMS button.")
except Exception:
print("No Send SMS page detected or an error occurred")
# Wait for the 2FA page
try:
print("Waiting for the 2FA page...")
wait.until(
EC.presence_of_element_located((By.XPATH, '//*[@id="sms_totp"]'))
)
print("2FA page loaded. Please complete the 2FA process manually.")
# Pause the script until the user completes the 2FA process
input("Complete the 2FA process and press Enter to continue...")
# Verification process should be completed by now
print("2FA process completed.")
except Exception:
print("No 2FA page detected or an error occurred")
try:
print("Waiting for authorization page to load...")
authorize_button = wait.until(
EC.element_to_be_clickable((By.NAME, 'authorize'))
)
authorize_button.click()
print("Clicked authorize button.")
except Exception:
print("An error occurred during GitHub authorization")
def wait_for_gallery_load():
try:
# Wait until the `window.loadGalleryComplete` flag is set to true
wait.until(lambda driver: driver.execute_script('return window.loadGalleryComplete === true;'))
except Exception as e:
print(f"Error waiting for gallery load: {str(e)}")
def main():
login_with_github()
# After logging in, navigate to the gallery page
driver.get("http://localhost/#/gallery")
try:
wait.until(EC.presence_of_element_located((By.CLASS_NAME, "MuiSelect-root")))
select = driver.find_element(By.CLASS_NAME, "MuiSelect-root")
select.click()
option = driver.find_element(By.XPATH, "//li[contains(text(), 'All Books')]")
option.click()
# Wait until the gallery page is loaded
wait.until(EC.presence_of_element_located((By.CSS_SELECTOR, "a[target='_blank'][href*='/editor?id=']")))
# Find all "Launch in Editor" buttons
buttons = driver.find_elements(By.CSS_SELECTOR, "a[target='_blank'][href*='/editor?id=']")
count = len(buttons)
print(f"Found {count} 'Launch in Editor' buttons.")
savecount = 0
# Click each "Launch in Editor" button and save
for i, button in enumerate(buttons):
if i == SKIP_COUNT - 1:
print(f"[{i + 1:3d}/{count}]: Skipping ")
continue
if i < SKIP_COUNT:
continue
if savecount >= FETCH_COUNT:
print(f"[{i + 1:3d}/{count}]: Skipping ")
break
button.click()
driver.switch_to.window(driver.window_handles[-1]) # Switch to the newly opened editor tab/window
try:
state = 0
# Wait for the page to load and the Save button to be clickable
save_button = wait.until(EC.element_to_be_clickable((By.XPATH, '//*[@id="root"]/div/header/div[2]/div[1]/button[2]')))
state += 1
# Wait for the gallery to fully load
wait_for_gallery_load()
state += 1
# Click the "Save" button
ActionChains(driver).move_to_element(save_button).click().perform()
state += 1
# Optionally, you can verify the snackbar message if needed
wait.until(EC.visibility_of_element_located((By.CLASS_NAME, "MuiSnackbar-root")))
state += 1
# Verify the share button is displayed
wait.until(EC.presence_of_element_located((By.XPATH, '//*[@id="root"]/div/header/div[1]/button[2]')))
state += 1
# Verify the "last saved" text is displayed
last_saved_text = wait.until(EC.presence_of_element_located((By.XPATH, '//*[@id="root"]/div/header/div[1]/p')))
state += 1
savecount += 1
print(f"[{i + 1:3d}/{count}]: {last_saved_text.text}")
except Exception:
print(f"[{i + 1:3d}/{count}]: Error while saving diagram {state}")
# Close the editor tab/window and switch back to the gallery
driver.close()
driver.switch_to.window(driver.window_handles[0])
# Keep the browser window open
except Exception as e:
print(f"Error while saving diagram: {str(e)}")
finally:
print(f"[{savecount:3d}/{count}] 'Launch in Editor' buttons have been clicked and saved.")
# Keep the browser open until manually closed
input("Press Enter to close the browser...")
driver.quit()
if __name__ == "__main__":
main()
|
