From 51001d66a58131ee63a68040abbafbf19e374870 Mon Sep 17 00:00:00 2001
From: Danny Robson <danny@nerdcruft.net>
Date: Sun, 14 Jul 2019 11:15:29 +1000
Subject: [PATCH] satisfactory: add typing to Cookbook member functions

---
 satisfactory/__init__.py | 14 +++++++-------
 1 file changed, 7 insertions(+), 7 deletions(-)

diff --git a/satisfactory/__init__.py b/satisfactory/__init__.py
index c76b4ed..9942279 100644
--- a/satisfactory/__init__.py
+++ b/satisfactory/__init__.py
@@ -1,7 +1,7 @@
 import os
 import json
 
-from typing import Dict
+from typing import Dict, Generator
 
 class Cookbook(object):
     __recipes: Dict[str, Dict]
@@ -16,23 +16,23 @@ class Cookbook(object):
                 with open(path, 'r') as src:
                     self.__recipes[name] = json.load(src)
 
-    def __getitem__(self, item: str):
+    def __getitem__(self, item: str) -> Dict[str, Dict]:
         return self.__recipes[item]
 
     def all(self):
         return self.__recipes.keys()
 
-    def recipes(self, name: str) -> dict:
+    def recipes(self, name: str) -> Dict:
         return self.__recipes[name]['recipes']
 
-    def is_component(self, name):
+    def is_component(self, name: str) -> bool:
         return 'component' in self.__recipes[name]['type']
 
-    def components(self):
+    def components(self) -> Generator[Dict, None, None]:
         return (i for i in self.all() if self.is_component(i))
 
-    def is_resource(self, name):
+    def is_resource(self, name: str) -> bool:
         return 'resource' in self.__recipes[name]['type']
 
-    def resources(self):
+    def resources(self) -> Generator[Dict, None, None]:
         return (i for i in self.all() if self.is_resource(i))