Added Category class

1 files changed, 13 insertions, 0 deletions

diff --git a/grumpy/models.py b/grumpy/models.py
index 4a74cbd..33eb6f6 100644
--- a/grumpy/models.py
+++ b/grumpy/models.py
@@ -25,6 +25,19 @@ package_herds = Table('package_herds', Base.metadata,
     Column('herd_id', Integer, ForeignKey('herds.id'))
     )
 
+class Category(Base):
+    """Represents portage categories"""
+
+    __tablename__ = 'categories'
+    id = Column('id', Integer, primary_key=True)
+    cat = Column('category', String, nullable=False, unique=True)
+
+    def __init__(self, cat):
+        self.cat = cat
+
+    def __repr__(self):
+        return '<%s> "%s"' % (self.__class__.__name__, self.cat)
+
 class Developer(Base):
     """Represents developers in the system"""