From 45685840accf2bfbbfcc176891e5072192285e42 Mon Sep 17 00:00:00 2001
From: Nicolas Joyard <nicolas@njoyard.fr>
Date: Sat, 15 Apr 2017 08:32:45 +0200
Subject: [PATCH] Ajout ActeurAmendement aux apis

---
 parlapi/graphql/schema.py | 9 +++++++++
 parlapi/rest/setup.py     | 6 ++++++
 2 files changed, 15 insertions(+)

diff --git a/parlapi/graphql/schema.py b/parlapi/graphql/schema.py
index 821cdbc..44f282d 100644
--- a/parlapi/graphql/schema.py
+++ b/parlapi/graphql/schema.py
@@ -7,6 +7,7 @@ from graphene_sqlalchemy import SQLAlchemyObjectType, SQLAlchemyConnectionField
 from ..models import (
     Acte as ActeModel,
     Acteur as ActeurModel,
+    ActeurAmendement as ActeurAmendementModel,
     ActeurDocument as ActeurDocumentModel,
     ActeurDossier as ActeurDossierModel,
     ActeurReunion as ActeurReunionModel,
@@ -45,6 +46,13 @@ class Acteur(SQLAlchemyObjectType):
         model = ActeurModel
 
 
+class ActeurAmendement(SQLAlchemyObjectType):
+    class Meta:
+        exclude_fields = ('search_vector',)
+        interfaces = (relay.Node,)
+        model = ActeurAmendementModel
+
+
 class ActeurDocument(SQLAlchemyObjectType):
     class Meta:
         exclude_fields = ('search_vector',)
@@ -214,6 +222,7 @@ schema = graphene.Schema(
     types=[
         Acte,
         Acteur,
+        ActeurAmendement,
         ActeurDocument,
         ActeurDossier,
         ActeurReunion,
diff --git a/parlapi/rest/setup.py b/parlapi/rest/setup.py
index 538612f..525949d 100644
--- a/parlapi/rest/setup.py
+++ b/parlapi/rest/setup.py
@@ -4,6 +4,7 @@ from .api import API
 from ..models import (
     Acte,
     Acteur,
+    ActeurAmendement,
     ActeurDocument,
     ActeurDossier,
     ActeurReunion,
@@ -41,6 +42,11 @@ def setup_an_api(app):
         description=u'Acteurs (ministres, parlementaires...)'
     )
 
+    api.endpoint(
+        ActeurAmendement,
+        hidden=True
+    )
+
     api.endpoint(
         ActeurDocument,
         hidden=True
-- 
2.26.2