From 19689600816cd66753796aa3ec6aa4ff23a3dfc4 Mon Sep 17 00:00:00 2001
From: hugo <hugo@saturncloud.io>
Date: Fri, 30 Sep 2022 14:08:51 -0700
Subject: [PATCH 1/2] wip

---
 examples/registering-users/README.md   |  5 ++
 examples/registering-users/register.py | 81 ++++++++++++++++++++++++++
 2 files changed, 86 insertions(+)
 create mode 100644 examples/registering-users/README.md
 create mode 100644 examples/registering-users/register.py

diff --git a/examples/registering-users/README.md b/examples/registering-users/README.md
new file mode 100644
index 00000000..f961eed6
--- /dev/null
+++ b/examples/registering-users/README.md
@@ -0,0 +1,5 @@
+# Registering Users
+
+This example illustrates how to write a script to register users in Saturn Cloud.
+
+This example takes a list of email addresses (passed in via an `EMAILS_FOR_ACCOUNTS` environment variable). It will check if a user account exists or not, and if not, create an account for the user. The reason we are passing this in via an ENV var is so that we can pass this via the Saturn secrets manager.
diff --git a/examples/registering-users/register.py b/examples/registering-users/register.py
new file mode 100644
index 00000000..6eec90fa
--- /dev/null
+++ b/examples/registering-users/register.py
@@ -0,0 +1,81 @@
+
+from urllib.parse import urlencode
+
+import requests
+
+
+# this should be populated by the secrets manager
+EMAILS_FOR_ACCOUNTS = os.getenv("EMAILS_FOR_ACCOUNTS")
+
+# this should be populated by Saturn. This script will only work if
+# run from an account that has admin access
+BASE_URL = os.getenv("BASE_URL")
+SATURN_TOKEN = os.getenv("SATURN_TOKEN")
+saturn_headers = {"Authorization": f"token {SATURN_TOKEN}"}
+
+def check_for_account_by_email(email: str) -> bool:
+    url = f"{BASE_URL}/api/users"
+    query_string = urlencode(dict(q=f"email:{email}", page=1, size=1))
+    url = url + "?" + query_string
+    response = requests.get(url, headers=saturn_headers)
+    results = response.json()['users']
+    if results:
+        return True
+    return False
+
+
+def check_for_account_by_username(username: str) -> bool:
+    url = f"{BASE_URL}/api/users"
+    query_string = urlencode(dict(q=f"username:{username}", page=1, size=1))
+    url = url + "?" + query_string
+    response = requests.get(url, headers=saturn_headers)
+    results = response.json()['users']
+
+    if results:
+        return True
+    return False
+
+
+def make_unique_username(email: str) -> str:
+    candidate_username = email.split('@')[0]
+    candidate_username = "".join(c for c in candidate_username if c.isalnum())
+
+    # we'll try 100 integers until we get a unique name
+    for c in range(100):
+        to_try = candidate_username
+        if c:
+            to_try = candidate_username + str(c)
+        if not check_for_account_by_username(to_try):
+            return to_try
+    raise ValueError(f'unable to find username for {candidate_username}')
+
+
+def make_account(username: str, email: str):
+    url = f"{BASE_URL}/api/users"
+    body = dict(
+        username=username,
+        email=email,
+        admin=False,
+        locked=False,
+        send_reset_email=False,
+        prevent_duplicate_emails=True,
+    )
+    response = requests.post(url, json=body, headers=saturn_headers)
+    print(response.json())
+
+
+def ensure_account_exists(email: str) -> None:
+    if check_for_account_by_email(email):
+        return
+    username = make_unique_username(email)
+    make_account(username, email)
+
+
+def run():
+    for email in EMAILS_FOR_ACCOUNTS.split('\n'):
+        if email:
+            ensure_account_exists(email)
+
+
+if __name__ == "__main__":
+    run()

From f630d3132aa5e88a8c6d18c1604a5309fc42a9f2 Mon Sep 17 00:00:00 2001
From: hhuuggoo <hugo@saturncloud.io>
Date: Fri, 30 Sep 2022 16:09:18 -0700
Subject: [PATCH 2/2] Update register.py

typo
---
 examples/registering-users/register.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/examples/registering-users/register.py b/examples/registering-users/register.py
index 6eec90fa..5230086a 100644
--- a/examples/registering-users/register.py
+++ b/examples/registering-users/register.py
@@ -1,4 +1,4 @@
-
+import os
 from urllib.parse import urlencode
 
 import requests