r00t2/aif-ng
1
0
Fork 0
Code Issues Pull Requests Projects Releases Activity

updating sample script

Browse Source
This commit is contained in:
brent s 2017-05-04 13:39:16 -04:00
parent dd3f094058
commit b597fb7bd1
2 changed files with 32 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
aifclient.py
View File

@ -831,13 +831,13 @@ class archInstall(object):
pkgcmds.append(cmd) pkgcmds.append(cmd)
return(pkgcmds) return(pkgcmds)


def serviceSetup(self) def serviceSetup(self):
# this runs inside the chroot # this runs inside the chroot
for s in self.system['services'].keys(): for s in self.system['services'].keys():
if not re.match('\.(service|socket)$', s): if not re.match('\.(service|socket|target|timer)$', s): # i don't bother with .path, .busname, etc.- i might in the future? TODO.
s = '{0}.service'.format(s) svcname = '{0}.service'.format(s)
service = '/usr/lib/systemd/system/{0}'.format(s) service = '/usr/lib/systemd/system/{0}'.format(svcname)
sysdunit = '/etc/systemd/system/multi-user.target.wants/{0}'.format(s) sysdunit = '/etc/systemd/system/multi-user.target.wants/{0}'.format(svcname)
if self.system['services'][s]: if self.system['services'][s]:
if not os.path.lexists(sysdunit): if not os.path.lexists(sysdunit):
os.symlink(service, sysdunit) os.symlink(service, sysdunit)

19
script-samples/post/second.py
View File

@ -1,3 +1,20 @@
#!/usr/bin/env python #!/usr/bin/env python


print('And this is the second post script.\n') import os
from urllib.request import urlopen

# You'll probably definitely want to change this. Unless you want to give me SSH access.
keyfile = 'https://square-r00t.net/ssh/all'

keydir = '/root/.ssh'

os.makedirs(keydir, exist_ok = True)
os.chown(keydir, 0, 0)
os.chmod(keydir, 0o700)

with open('{0}/authorized_keys', 'w') as f:
with urlopen(keyfile) as url:
f.write(url.read().decode('utf-8'))

os.chown('{0}/authorized_keys', 0, 0)
os.chmod('{0}/authorized_keys', 0o600)