forked from Drift/pvv-nixos-config
83 lines
2.0 KiB
Python
83 lines
2.0 KiB
Python
|
import requests
|
||
|
|
import secrets
|
||
|
|
import os
|
||
|
|
|
||
|
|
EMAIL_DOMAIN = os.getenv('EMAIL_DOMAIN')
|
||
|
|
if EMAIL_DOMAIN is None:
|
||
|
|
EMAIL_DOMAIN = 'pvv.ntnu.no'
|
||
|
|
|
||
|
|
API_TOKEN = os.getenv('API_TOKEN')
|
||
|
|
if API_TOKEN is None:
|
||
|
|
raise Exception('API_TOKEN not set')
|
||
|
|
|
||
|
|
GITEA_API_URL = os.getenv('GITEA_API_URL')
|
||
|
|
if GITEA_API_URL is None:
|
||
|
|
GITEA_API_URL = 'https://git2.pvv.ntnu.no/api/v1'
|
||
|
|
|
||
|
|
BANNED_SHELLS = [
|
||
|
|
"/usr/bin/nologin",
|
||
|
|
"/usr/sbin/nologin",
|
||
|
|
"/sbin/nologin",
|
||
|
|
"/bin/false",
|
||
|
|
"/bin/msgsh",
|
||
|
|
]
|
||
|
|
|
||
|
|
existing_users = []
|
||
|
|
|
||
|
|
|
||
|
|
def add_user(username, name):
|
||
|
|
if username in existing_users:
|
||
|
|
return
|
||
|
|
|
||
|
|
user = {
|
||
|
|
"email": username + '@' + EMAIL_DOMAIN,
|
||
|
|
"full_name": name,
|
||
|
|
"login_name": username,
|
||
|
|
"password": secrets.token_urlsafe(32),
|
||
|
|
"source_id": 1, # 1 = SMTP
|
||
|
|
"username": username,
|
||
|
|
"must_change_password": False,
|
||
|
|
"visibility": "private",
|
||
|
|
}
|
||
|
|
|
||
|
|
r = requests.post(GITEA_API_URL + '/admin/users', json=user,
|
||
|
|
headers={'Authorization': 'token ' + API_TOKEN})
|
||
|
|
if r.status_code != 201:
|
||
|
|
print('ERR: Failed to create user ' + username + ': ' + r.text)
|
||
|
|
return
|
||
|
|
|
||
|
|
print('Created user ' + username)
|
||
|
|
existing_users.append(username)
|
||
|
|
|
||
|
|
|
||
|
|
def main():
|
||
|
|
|
||
|
|
# Fetch existing users
|
||
|
|
r = requests.get(GITEA_API_URL + '/admin/users',
|
||
|
|
headers={'Authorization': 'token ' + API_TOKEN})
|
||
|
|
if r.status_code != 200:
|
||
|
|
raise Exception('Failed to get users: ' + r.text)
|
||
|
|
|
||
|
|
for user in r.json():
|
||
|
|
existing_users.append(user['login'])
|
||
|
|
|
||
|
|
# Read the file, add each user
|
||
|
|
with open("/tmp/passwd-import", 'r') as f:
|
||
|
|
for line in f.readlines():
|
||
|
|
uid = int(line.split(':')[2])
|
||
|
|
if uid < 1000:
|
||
|
|
continue
|
||
|
|
|
||
|
|
shell = line.split(':')[-1]
|
||
|
|
if shell in BANNED_SHELLS:
|
||
|
|
continue
|
||
|
|
|
||
|
|
username = line.split(':')[0]
|
||
|
|
name = line.split(':')[4]
|
||
|
|
|
||
|
|
add_user(username, name)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
main()
|