1 files changed, 5 insertions, 2 deletions
diff --git a/readyaml.py b/readyaml.py
index b88d1e0..7008263 100755
--- a/readyaml.py
+++ b/readyaml.py
@@ -5,11 +5,13 @@ updates the database with the contents."""
 import argparse
 import sqlite3
 import sys
+import typing
 
 from debian.debian_support import version_compare
 import yaml
 
-def readyaml(db, stream):
+
+def readyaml(db: sqlite3.Connection, stream: typing.TextIO) -> None:
     cur = db.cursor()
     cur.execute("PRAGMA foreign_keys = ON;")
     gen = yaml.load_all(stream, yaml.CSafeLoader)
@@ -53,7 +55,8 @@ def readyaml(db, stream):
                          for func, hexhash in entry["hashes"].items()))
     raise ValueError("missing commit block")
 
-def main():
+
+def main() -> None:
     parser = argparse.ArgumentParser()
     parser.add_argument("-d", "--database", action="store",
                         default="test.sqlite3",