refactor loading configs

pyfake/adjective.py

@@ -1,14 +1,15 @@
 from .generator import Generator
+from pathlib import Path
 import random
 import yaml
 
+with open(str(Path(__file__).parent) + '/config/adjectives.yml', 'r') as file:
+    CONFIG = yaml.safe_load(file)
 
 class Adjective(Generator):
-    with open("config/adjectives.yml", "r") as file:
-        CONFIG = yaml.safe_load(file)
-
     def adjective(self):
-        return random.choice(self.CONFIG["adjectives"])
+        print(CONFIG)
+        return random.choice(CONFIG)
 
     def generate(self):
         self.adjective = self.adjective()

pyfake/book.py

@@ -1,8 +1,9 @@
 from .generator import Generator
+from pathlib import Path
 import random
 import yaml
 
-with open('config/books.yml', 'r') as file:
+with open(str(Path(__file__).parent) + '/config/books.yml', 'r') as file:
     CONFIG = yaml.safe_load(file)
 
 class Book(Generator):

pyfake/company.py

@@ -1,16 +1,17 @@
 from .generator import Generator
+from pathlib import Path
 import random
 import yaml
 
-class Company(Generator):
-    with open('config/company.yml', 'r') as file:
-        CONFIG = yaml.safe_load(file)
+with open(str(Path(__file__).parent) + '/config/company.yml', 'r') as file:
+    CONFIG = yaml.safe_load(file)
 
+class Company(Generator):
     def company_name(self):
-        return random.choice(self.CONFIG['names'])
+        return random.choice(CONFIG['names'])
 
     def company_suffix(self):
-        return random.choice(self.CONFIG['suffixes'])
+        return random.choice(CONFIG['suffixes'])
 
     def generate(self):
         self.size = random.randint(1, 100)

pyfake/config/adjectives.yml

@@ -0,0 +1,57 @@
+---
+- arrogant
+- beautiful
+- best
+- biggest
+- circular
+- competent
+- confusing
+- courageous
+- cumbersome
+- dainty
+- dark
+- devoted
+- efficient
+- expensive
+- fickle
+- generous
+- graceful
+- grumpy
+- happiest
+- helpful
+- hopeful
+- hot
+- hungry
+- mild
+- modern
+- muscular
+- new
+- new
+- oldest
+- oval
+- petite
+- prettiest
+- proud
+- rich
+- rocky
+- round
+- round
+- royal
+- rude
+- safe
+- scary
+- sleepy
+- small
+- smart
+- smelly
+- stable
+- strange
+- strange
+- tardy
+- thin
+- weary
+- wicked
+- witty
+- wooden
+- youthful
+- zany

pyfake/country.py

@@ -1,13 +1,14 @@
 from .generator import Generator
+from pathlib import Path
 import yaml
 import random
 
-class Country(Generator):
-    with open('config/countries.yml', 'r') as file:
-        CONFIG = yaml.safe_load(file)
+with open(str(Path(__file__).parent) + '/config/countries.yml', 'r') as file:
+    CONFIG = yaml.safe_load(file)
 
+class Country(Generator):
     def generate(self):
-        data = random.choice(self.CONFIG)
+        data = random.choice(CONFIG)
 
         self.name         = data['name']
         self.currency     = data['currency']

pyfake/person.py

@@ -1,13 +1,14 @@
 from .generator import Generator
+from pathlib import Path
 import random
 import yaml
 
-class Person(Generator):
-    with open('config/people.yml', 'r') as file:
-        CONFIG = yaml.safe_load(file)
+with open(str(Path(__file__).parent) + '/config/people.yml', 'r') as file:
+    CONFIG = yaml.safe_load(file)
 
+class Person(Generator):
     def generate(self):
-        data = random.choice(self.CONFIG)
+        data = random.choice(CONFIG)
 
         self.first_name = data['first_name']
         self.last_name  = data['last_name']

pyfake/sport.py

@@ -1,13 +1,14 @@
 from .generator import Generator
+from pathlib import Path
 import yaml
 import random
 
-class Sport(Generator):
-    with open('config/sports.yml', 'r') as file:
-        CONFIG = yaml.safe_load(file)
+with open(str(Path(__file__).parent) + '/config/sports.yml', 'r') as file:
+    CONFIG = yaml.safe_load(file)
 
+class Sport(Generator):
     def generate(self):
-        data = random.choice(self.CONFIG)
+        data = random.choice(CONFIG)
 
         self.name         = data['name']
         self.players      = data['players']

tests/test_adjective.py

@@ -3,4 +3,4 @@
 def test_adjective():
     adj = Adjective().adjective
 
-    assert adj in Adjective().CONFIG['adjectives']
+    assert isinstance(adj, str)