raannakasturi commited on
Commit
208023b
·
1 Parent(s): a90db02

Refactor getTokenCert.py to remove redundant code

Browse files
Files changed (2) hide show
  1. getTokenCert.py +0 -1
  2. main.py +12 -7
getTokenCert.py CHANGED
@@ -55,7 +55,6 @@ def finalize_order(client, order, deadline):
55
  data = client.poll_and_finalize(order, deadline=deadline)
56
  return data
57
  except Exception as e:
58
- data = client.poll_and_finalize(order, deadline=deadline)
59
  print(f"Error finalizing order: {e}")
60
  return None
61
 
 
55
  data = client.poll_and_finalize(order, deadline=deadline)
56
  return data
57
  except Exception as e:
 
58
  print(f"Error finalizing order: {e}")
59
  return None
60
 
main.py CHANGED
@@ -127,14 +127,14 @@ def main(i_domains, wildcard, email, ca_server, key_type, key_size=None, key_cur
127
  cf_non_wildcard(verification_tokens, email, exchange)
128
  except Exception as e:
129
  print(f"Error adding TXT records: {e}")
130
- for i in range(30):
131
  print(f"Waiting for {30-i} seconds", end="\r")
132
  time.sleep(1)
133
  retries = 0
134
  while not verify_txt_records(verification_tokens, exchange):
135
  print("TXT records not verified yet")
136
  retries = retries+1
137
- if retries >=50:
138
  break
139
  time.sleep(5)
140
  cert = verify_tokens(pgk_client, challs, order)
@@ -145,9 +145,14 @@ def main(i_domains, wildcard, email, ca_server, key_type, key_size=None, key_cur
145
  print("TXT records deleted successfully")
146
  except Exception as e:
147
  print(f"Error deleting TXT records or no TXT records exist: {e}")
148
- private_key = private_key.decode("utf-8")
149
- csr = csr.decode("utf-8")
150
- cert = cert.decode("utf-8")
 
 
 
 
 
151
  generation_details = f"""
152
  SSL Certificate for {i_domains} were generated successfully, using Project Gatekeeper, a free SSL Certificate creator tool.
153
  SSL Provider = {ca_server}
@@ -164,8 +169,8 @@ def main(i_domains, wildcard, email, ca_server, key_type, key_size=None, key_cur
164
  return private_key, csr, cert, email_status
165
 
166
  if __name__ == "__main__":
167
- DOMAINS = 'thenayankasturi.eu.org'
168
- ca_server = "Google" #Let's Encrypt (Testing), Let's Encrypt, Google (Testing), Google, Buypass (Testing), Buypass, ZeroSSL, SSL.com
169
  EMAIL = "[email protected]"
170
  key_type = "ecc"
171
  key_curve = "ec384"
 
127
  cf_non_wildcard(verification_tokens, email, exchange)
128
  except Exception as e:
129
  print(f"Error adding TXT records: {e}")
130
+ for i in range(10):
131
  print(f"Waiting for {30-i} seconds", end="\r")
132
  time.sleep(1)
133
  retries = 0
134
  while not verify_txt_records(verification_tokens, exchange):
135
  print("TXT records not verified yet")
136
  retries = retries+1
137
+ if retries >=3:
138
  break
139
  time.sleep(5)
140
  cert = verify_tokens(pgk_client, challs, order)
 
145
  print("TXT records deleted successfully")
146
  except Exception as e:
147
  print(f"Error deleting TXT records or no TXT records exist: {e}")
148
+ try:
149
+ private_key = private_key.decode("utf-8")
150
+ csr = csr.decode("utf-8")
151
+ cert = cert.decode("utf-8")
152
+ except Exception as e:
153
+ private_key = "Failed to obtain private key"
154
+ csr = "Failed to obtain CSR"
155
+ cert = "Failed to obtain certificate"
156
  generation_details = f"""
157
  SSL Certificate for {i_domains} were generated successfully, using Project Gatekeeper, a free SSL Certificate creator tool.
158
  SSL Provider = {ca_server}
 
169
  return private_key, csr, cert, email_status
170
 
171
  if __name__ == "__main__":
172
+ DOMAINS = 'nayankasturi.eu.org'
173
+ ca_server = "Let's Encrypt (Testing)" #Let's Encrypt (Testing), Let's Encrypt, Google (Testing), Google, Buypass (Testing), Buypass, ZeroSSL, SSL.com
174
  EMAIL = "[email protected]"
175
  key_type = "ecc"
176
  key_curve = "ec384"