Resolves #10 using playhouse generic foreign keys

models/authority.py

@@ -1,12 +1,13 @@
 #!/usr/bin/env python3
 # -*- coding: utf-8 -*-
 
-from peewee import *
+from playhouse.gfk import *
 
 import os
 import os.path
 
 from models import customModel
+from models.certificate import Certificate
 
 from paths import *
 
@@ -16,6 +17,8 @@ Module of base classes to handle authorities
 
 class Authority(customModel.CustomModel):
 
+    signed_certificates = ReverseGFK(Certificate, 'authority_type', 'authority_id')
+
     request_allowed = []
 
     # data stored in the database

models/certificate.py

@@ -1,14 +1,13 @@
 #!/usr/bin/env python3
 # -*- coding: utf-8 -*-
 
-from peewee import *
+from playhouse.gfk import *
 
 import os
 import json
 
 from models import customModel
 
-from models.authority import Authority
 from paths import *
 
 
@@ -16,10 +15,9 @@ class Certificate(customModel.CustomModel):
     """
     """
 
-    signed_by = ForeignKeyField(
-            Authority,
-            related_name = 'signed_certificates',
-            )
+    authority_type = CharField(null=True)
+    authority_id = IntegerField(null=True)
+    authority = GFKField('authority_type', 'authority_id')
 
     cert_id = CharField(
                 index = True,

models/customModel.py

@@ -1,5 +1,5 @@
 from paths import *
-from peewee import *
+from playhouse.gfk import *
 import os
 
 custom_db = SqliteDatabase(os.path.join(MANAGER_PATH, 'ca_manager.db'))

models/ssh.py

@@ -1,7 +1,7 @@
 #!/usr/bin/env python3
 # -*- coding: utf-8 -*-
 
-from peewee import *
+from playhouse.gfk import *
 
 from datetime import datetime
 import os.path
@@ -103,7 +103,7 @@ class SSHAuthority(Authority):
         pub_key_path = request.destination
         
         cert = Certificate(
-                signed_by = self,
+                authority = self,
                 cert_id = request.req_id,
                 date_issued = datetime.now(),
                 receiver = request.receiver,

models/ssl.py

@@ -1,7 +1,7 @@
 #!/usr/bin/env python3
 # -*- coding: utf-8 -*-
 
-from peewee import *
+from playhouse.gfk import *
 
 import os
 import os.path