Mercurial > code > home > repos > infra

comparison system.py @ 1:b664f1027992

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
system.py port from ansible
author drewp@bigasterisk.com
date Thu, 04 Nov 2021 18:10:02 -0700
parents
children 7f7af7e2ba8d
comparison
equal deleted inserted replaced
0:1550a6db59b3 1:b664f1027992
1 import os
2 from pyinfra import host
3 from pyinfra.operations import server, files, apt, ssh, systemd
4 from pyinfra.facts.server import LinuxDistribution
5
6 is_pi = host.get_fact(LinuxDistribution)['name'] == 'Debian'
7 is_wifi_pi = host.name in ['frontdoor', 'living']
8
9 TZ = 'America/Los_Angeles'
10
11 #
12 # system
13 #
14
15 server.hostname(hostname=host.name)
16 files.link(path='/etc/localtime', target=f'/usr/share/zoneinfo/{TZ}')
17 files.replace(path='/etc/timezone', match='.*', replace=TZ)
18 apt.packages(update=True,
19 cache_time=86400,
20 packages=['tzdata'],
21 force=True,
22 env={
23 'TZ': TZ,
24 'LANG': 'en_US.UTF-8',
25 'DEBIAN_FRONTEND': 'noninteractive'
26 })
27
28 #
29 # fstab
30 #
31
32 fstab_file = f'files/{host.name}_fstab'
33 if os.path.exists(fstab_file):
34 files.put(src=fstab_file, dest='/etc/fstab')
35 if is_pi:
36 for line in [
37 'tmpfs /var/log tmpfs defaults,noatime,mode=0755 0 0',
38 'tmpfs /tmp tmpfs defaults,noatime 0 0',
39 ]:
40 files.line(path="/etc/fstab", line=line, replace=line)
41
42 # stop SD card corruption (along with some mounts in fstab)
43 apt.packages(packages=['dphys-swapfile'], present=False)
44
45 #
46 # pkgs
47 #
48
49 if not is_pi:
50 apt.key(keyserver='keyserver.ubuntu.com', keyid='8B48AD6246925553')
51
52 if is_pi:
53 apt.packages(packages=['dirmngr', 'gnupg2', 'apt-utils'])
54 files.template(src='templates/pi_sources.list.j2', dest='/etc/apt/sources.list', rel='buster')
55
56 apt.key(src='https://ftp-master.debian.org/keys/archive-key-8.asc')
57 apt.key(src='https://ftp-master.debian.org/keys/archive-key-8-security.asc')
58 apt.key(src='https://ftp-master.debian.org/keys/archive-key-9-security.asc')
59
60 files.file(path='/etc/apt/sources.list.d/raspi.list', present=False)
61
62 if is_wifi_pi:
63 files.put(dest="/etc/network/interfaces.d/wlan0", src="files/pi_wlan0_powersave")
64 ssh.command(host.name, "iw wlan0 set power_save off")
65
66 # see https://www.raspberrypi.org/documentation/configuration/config-txt/memory.md#:~:text=txt-,gpu_mem,-Specifies
67 # to port to pyinfra
68 #- name: unused display; give ram to OS
69 # lineinfile: dest=/boot/config.txt line="gpu_mem=16" regexp="^gpu_mem="
70 # when: "'with_x11' not in group_names"
71
72 # for beacon
73 #enable_uart=1
74 #dtoverlay=pi3-miniuart-bt
75 #core_freq=250
76
77 # for tiny_screen
78 #to port to pyinfra
79 #- lineinfile: dest=/boot/config.txt line="dtparam=spi=on" regexp="^dtparam=spi="
80
81 # i hope this is deletable
82 # downgrade strictness so I can install from https://archive.raspberrypi.org/
83 # https://bugs.debian.org/cgi-bin/bugreport.cgi?bug=907788
84 #- lineinfile: dest=/etc/ssl/openssl.cnf line="#CipherString = DEFAULT@SECLEVEL=2" regexp="CipherString ?="
85
86 # may be fixed in k3s, not sure
87 # raspbian defaults to `iptables -V` -> iptables v1.8.4 (nf_tables), which won't work with k3s
88 # - command: update-alternatives --set iptables /usr/sbin/iptables-legacy
89 if not is_pi:
90 apt.key(src='https://dl.google.com/linux/linux_signing_key.pub')
91 apt.repo(src='deb [arch=amd64] http://dl.google.com/linux/chrome/deb/ stable main')
92
93 apt.key(src='https://packages.microsoft.com/keys/microsoft.asc')
94 apt.repo(src="deb [arch=amd64,arm64,armhf] http://packages.microsoft.com/repos/code stable main")
95
96 apt.ppa(src="ppa:savoury1/blender")
97
98 apt.key(keyserver='keyserver.ubuntu.com', keyid='F24AEA9FB05498B7')
99 apt.repo(src="deb [arch=amd64,i386] https://repo.steampowered.com/steam/ stable steam")
100
101 if is_pi:
102 apt.key(src="https://download.docker.com/linux/raspbian/gpg")
103 apt.repo(src="deb [arch=armhf] https://download.docker.com/linux/raspbian stretch stable")
104 apt.repo(src='deb http://deb.debian.org/debian/ unstable main') # maybe for WG
105
106 # don't try to get aufs-dkms on rpi-- https://github.com/docker/for-linux/issues/709
107 apt.packages(packages=['docker.io'], no_recommends=True)
108
109 apt.packages(packages=[
110 'build-essential',
111 'python3-pip',
112 'python3-virtualenv',
113 'python3-docker',
114 'rsync',
115 'i2c-tools',
116 'sysstat',
117 ])
118
119 #
120 # ssh
121 #
122
123 systemd.service(
124 service='ssh',
125 running=True,
126 enabled=True,
127 )
128
129 files.line(path='/etc/ssh/ssh_config', line="HashKnownHosts", replace="HashKnownHosts no")
130
131 if is_pi:
132 auth_keys = '/home/pi/.ssh/authorized_keys'
133 files.file(path=auth_keys, user='pi', group='pi', mode=600)
134 for pubkey in [
135 'ecdsa-sha2-nistp256 AAAAE2VjZHNhLXNoYTItbmlzdHAyNTYAAAAIbmlzdHAyNTYAAABBBNlR7hereUHqw/RHQau0F7+vQZKAxduM+SD4R76FhC+4Zi078Pv04ZLe9qdM/NBlB/grLGhG58vaGmnWPpJ3QJs= drewp@plus',
136 'ecdsa-sha2-nistp256 AAAAE2VjZHNhLXNoYTItbmlzdHAyNTYAAAAIbmlzdHAyNTYAAABBBOR+iV8Qm/rAfmq0epXYfnp5ZTfBl8eidFzw1GmyZ3fPUFAshWn839fQ5DPj9xDPtMy9kTtrB5bK1SnynFzDqzQ= drewp@bang',
137 ]:
138 files.line(path=auth_keys, line=pubkey, replace=pubkey)
139
140 #
141 # docker (delete this?)
142 #
143
144 files.put(src='files/docker_daemon.json', dest='/etc/docker/daemon.json')
145 systemd.service(service='docker', running=True, enabled=True, restarted=True)
146
147 if not is_pi:
148 files.line(path='/etc/update-manager/release-upgrades', line="^Prompt=", replace="Prompt=normal")
149
150 files.line(path='/etc/ssh/sshd_config', line="^UseDNS\b", replace="UseDNS no")
151 systemd.service(service='sshd', reloaded=True)
152
153 #
154 # special hosts
155 #
156
157 if host.name == "bang":
158 apt.packages(packages=[
159 'libzfs2linux',
160 'zfsutils-linux',
161 'zfs-zed',
162 'zfs-auto-snapshot',
163 ])
164
165 # This is usable on pi where we don't care when they reboot:
166 #- name: apt_upgrade
167 # apt: upgrade=full
168 #- name: Check if a reboot is required
169 # register: file
170 # stat: path=/var/run/reboot-required get_md5=no
171 #- name: Reboot the server
172 # command: /sbin/reboot
173 # when: file.stat.exists == true