feat: added new badge model (amundsen-io#55)

* added new badge model Signed-off-by: Allison Suarez Miranda <asuarezmiranda@lyft.com> * changed type to category Signed-off-by: Allison Suarez Miranda <asuarezmiranda@lyft.com> * bumped version and updated fields Signed-off-by: Allison Suarez Miranda <asuarezmiranda@lyft.com> * forgot the patch Signed-off-by: Allison Suarez Miranda <asuarezmiranda@lyft.com>
dorianj · Aug 25, 2020 · 09897d9 · 09897d9
1 parent 2c8b494
commit 09897d9
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 1 deletion.
diff --git a/amundsen_common/models/table.py b/amundsen_common/models/table.py
@@ -33,6 +33,19 @@ class Meta:
         register_as_scheme = True
 
 
+@attr.s(auto_attribs=True, kw_only=True)
+class Badge:
+    badge_name: str = attr.ib()
+    category: str = attr.ib()
+    badge_type: str = attr.ib()
+
+
+class BadgeSchema(AttrsSchema):
+    class Meta:
+        target = Badge
+        register_as_scheme = True
+
+
 @attr.s(auto_attribs=True, kw_only=True)
 class Watermark:
     watermark_type: Optional[str] = None

diff --git a/setup.py b/setup.py
@@ -5,7 +5,7 @@
 
 setup(
     name='amundsen-common',
-    version='0.3.6',
+    version='0.4.0',
     description='Common code library for Amundsen',
     long_description=open('README.md').read(),
     long_description_content_type='text/markdown',