optools/aif/scripts/post/sshkeys.py
2017-09-28 01:24:51 -04:00

31 lines
901 B
Python

#!/usr/bin/env python3
import os
import pwd
from urllib.request import urlopen
keysfile = 'https://square-r00t.net/ssh/all'
def copyKeys(keystring, user = 'root'):
uid = pwd.getpwnam(user).pw_uid
gid = pwd.getpwnam(user).pw_gid
homedir = os.path.expanduser('~{0}'.format(user))
sshdir = '{0}/.ssh'.format(homedir)
authfile = '{0}/authorized_keys'.format(sshdir)
with open(authfile, 'a') as f:
f.write(keystring)
os.makedirs(sshdir, mode = 0o700, exist_ok = True)
for basedir, dirs, files in os.walk(sshdir):
os.chown(basedir, uid, gid)
os.chmod(basedir, 0o700)
for f in files:
os.chown(os.path.join(basedir, f), uid, gid)
os.chmod(os.path.join(basedir, f), 0o600)
return()
def main():
with urlopen(keysfile) as keys:
copyKeys(keys.read().decode('utf-8'))
if __name__ == '__main__':
main()