From e07e64dfa5c6bea32da4c4ce645795cb1ff85209 Mon Sep 17 00:00:00 2001
From: Andrew Ekstedt <andrew.ekstedt@gmail.com>
Date: Tue, 5 Jun 2012 15:15:40 -0700
Subject: [PATCH] Display primary keys in table docs.

---
 pokedex/doc/tabledoc.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/pokedex/doc/tabledoc.py b/pokedex/doc/tabledoc.py
index ebd0f03..a24c82d 100644
--- a/pokedex/doc/tabledoc.py
+++ b/pokedex/doc/tabledoc.py
@@ -139,6 +139,10 @@ def generate_table_header(cls, remaining_attrs):
         pass
     yield u''
 
+    yield u'Primary key: %s.' % u', '.join(
+        u'**%s**' % col.key for col in cls.__table__.primary_key.columns)
+    yield u''
+
 def generate_common(cls, remaining_attrs):
     common_col_headers = []
     for c in cls.__table__.c: