49. Vaccinationsboten - ett miniprojekt
Jag har länge velat göra något enklare programmeringsprojekt som inte är kopplat till mina studier. Såg en Twitter-bot som kvittrade ut en förloppsindikator för hur många procent av USA:s befolkning som hunnit vaccineras mot covid-19 och tänkte att det vore nice att göra något liknande för Sveriges vaccinationsstatistik.
Jag kan inte längre hitta den ursprungliga boten men såg den här som gör något liknande för världens befolkning som helhet och den här för Storbritannien. För Sveriges del har jag hittills försökt hålla mig uppdaterad om statistiken genom att kika på SVT:s snygga grafiska representation av andelen vaccinerade. Den bygger på data från Folkhälsomyndigheten, som i sin tur hämtar den från Nationella vaccinationsregistret.
Skaffa Twitter Developer-konto
Jag har aldrig skapat någon Twitter-bot tidigare och började med att skaffa ett developer-konto på Twitter. Man måste ge en kort beskrivning av sitt projekt och vilka funktioner man tänkte använda, men det går väldigt snabbt och verkar godkännas på en gång. Jag gjorde först misstaget att ansöka om ett developer-konto kopplat till mitt personliga Twitter-konto, men eftersom jag ville ha ett separat konto för boten skaffade jag det och ansökte sedan på nytt med just det kontot.
Hämta data från Folkhälsomyndigheten
Den första uppgiften för projektet är att hämta data från Folkhälsomyndigheten. På deras hemsida presenteras antalet och andelen personer (>=18 år) som vaccinerats så här:
Folkhälsomyndighetens tabell över antalet och andelen vaccinerade med 1:a och 2:a dosen bland personer 18 år eller äldre.
Tyvärr finns det också mycket annat oväsentligt på sidan: sidomenyer, brödtext, länkar, andra tabeller, m.m. Jag vill bara hämta informationen från den här tabellen. Konsten att hämta ut information av intresse på nätet kallas data- eller webbskrapning (eng. web scraping). De tutorials jag hittat börjar alla med tipset att väja “Inspect”-alternativet när man högerklickar på sidan och undersöka hur datat är strukturerat.
I Firefox heter "Inspect"-alternativet i högerklicksmenyn "Inspect Element".
Då dyker det upp en sån här Inspector-ruta som visar HTML- och CSS-koden för sidan. Efter att ha klickat mig runt i myllret av <div>
-taggar hittade jag till slut tabellen:
Den nästlade HTLM-koden för tabellen av intresse i Inspector-fönstret.
Med vetskapen om att en tabell i HTML inleds med en <table>
-tag med en beskrivande caption som har kolumnrubriker inom <th>
-taggar och det egentliga datat inom <tr>
-taggar strider vi till verket och skriver ett Python-script för extrahera det. Jag följde den här steg-för-steg-guiden med några smärre ändringar för hämta rätt tabell och parsa siffrorna så som jag ville ha dem. HTML-parsningen sker med BeautifulSoup
-paketet och själva datat sparas i datamanipuleringsprogrammet pandas DataFrame
-struktur.
Det extraherade tabelldatat i pandas DataFrame-format.
Generera Twitter-meddelandets sträng
Jag försökte till en början få till förloppsindikatorerna med hjälp av tqdm
precis som killen i den här guiden hade gjort. Men när det inte blev bra valde jag istället en mer basic lösning, nämligen att själv skriva ut en indikator med hjälp av klammertecken, blockelement i unicode och punkter. Hur stabil den här lösningen är återstår att se, men det verkar fungera hjälpligt. Nackdelen är att indikatorerna inte får samma storlek när de twittras ut:
De handskrivna förloppsindikatorernas utseende i terminalen.
De handskrivna förloppsindikatorernas utseende på Twitter.
Uppdatering (20-04-2021): Hade helt missat att inte bara längden på förloppsindikatorn blev skev, utan också procentenhetens längd. Indikatorn med 20,4 % ser ju ut att vara nästan 50 %! Detta beror på att blockelementets storlek är större än punkterna, vilket inte syns i terminalutskriften men däremot på Twitter. Har nu löst detta genom att ersätta både procentblocket och punkterna med blockelement av samma storlek:
Utseendet hos de nya förloppsindikatorerna med blockelement av samma storlek.
Automatisera boten med GitHub Actions
För automatiseringen av boten drog jag stor nytta av den här guiden av Patrick Bos. Den introducerade mig till något som jag faktiskt var helt obekant med tidigare: GitHub Actions. Med hjälp av en YAML-fil som man lägger in under .github/workflows/
i repon kan man trigga GitHub att köra koden utifrån olika händelser (t.ex. att någon pushar till ett repo) eller schemalagda tidpunkter. Eftersom FOHM uppdaterar sin statistik varje dag från tisdag till fredag vid 14-tiden valde jag att köra programmet de dagarna vid halv 3. Notera att tidpunkten måste anges i cron-syntax och UCT-tid.
name: Tweet latest vaccination statistics for Sweden
on:
schedule:
- cron: '30 12 * * 2-5' # Run every Tuesday-Friday at 12:30 UTC time (14:30 Swedish time)
# Allows you to run this workflow manually from the Actions tab
workflow_dispatch:
jobs:
update:
runs-on: ubuntu-latest
steps:
- uses: actions/checkout@v2
- uses: actions/setup-python@v2
with:
python-version: '3.9.1'
- name: install dependencies
run: pip install -r requirements.txt
- name: run script
env:
BOT_API: $
BOT_API_SECRET: $
BOT_ACCESS: $
BOT_ACCESS_SECRET: $
run: python run.py
I YAML-filen specificerar Bos också vilka andra program som koden är beroende av (genom att länka till en requirements.txt
-fil) så att de kan installeras innan koden körs. Här initierar han också de miljövarialber, env
, som behövs för Twitter-kontots API (api-nyckel, access-token, etc.) och som i sin tur göms undan i “Secrets”-delen av GitHub (en annan ny komponent för mig!). Den hittar man genom att gå till Settings
–> Secrets
.
I programkoden importeras en speciell argumentparsare - configargparse
- som kan användas för att dels fånga upp miljövariablerna som anges i YAML-filen (och som sparas krypterade i Secrets
), men även variabler som anges i en separat konfigureringsfil. Detta gör det enklare att testa koden både i molnet och lokalt eftersom man då enkelt kan komma åt API-nycklarna på hemmadatorn genom den separata filen (som man skyddar från att pusha med .gitignore
).
En tredje och sista viktig komponent för att köra koden är Tweepy, ett Python-bibliotek för att kommunicera med Twitters API. Det går att se hur Bos kombinerar den med configargparse
här. Och i den här guiden finns ytterligare information för hur man kan använda Tweepy för fler ändamål än att bara skriva tweets.
Configargparse-parametrarna samt autentisering med Tweepy för att skapa ett API-objekt.
Slutresultatet
Boten heter @vaccination_SE och källkoden går att hitta här. Även om det inte var ett superavancerat projekt är jag ändå glad över att ha klarat av det och känner att jag lärt mig mycket på vägen. Det ska bli spännande att se hur väl GitHub Actions fungerar de närmaste dagarna och om det dyker upp några buggar.
Vaccinationsbotens slutgiltiga utseende.
Uppdatering (20-04-2021): Boten kördes på autopilot för första gången idag och det funkade! :D Men körningen skedde först 12 min efter utsatt tid, vilket tydligen är ganska vanligt. För att kompensera fördröjningen tänkte jag tidigarelägga schemat med 10 min.
Uppdatering (11-08-2021): Fyra månader senare fick jag mitt första runtime error när FHM plötsligt la till en punkt i slutet av tabellbeskrivningen, vilket gjorde att find_parent()
-metoden i BeautifulSoup
-paketet inte längre kunde hitta tabellen utifrån caption-taggen. Fixade det genom att lägga till den saknade punkten i koden, men det hela visar ändå på hur bräckligt webbskrapning kan vara inför ändringar på webbsidan.