Only attempt to count versions for versioned models.

This commit is contained in:
Lance Edgar 2015-02-10 11:23:35 -06:00
parent 75729be79f
commit 5ad5cb569d

View file

@ -34,6 +34,7 @@ except ImportError:
import sqlalchemy as sa import sqlalchemy as sa
from sqlalchemy_continuum import transaction_class, version_class from sqlalchemy_continuum import transaction_class, version_class
from sqlalchemy_continuum.utils import is_versioned
from rattail.db import model from rattail.db import model
from rattail.db.model.continuum import count_versions, model_transaction_query from rattail.db.model.continuum import count_versions, model_transaction_query
@ -169,9 +170,9 @@ class CrudView(View):
pass pass
def template_kwargs(self, form): def template_kwargs(self, form):
if form.creating: if not form.creating and is_versioned(self.mapped_class):
return {}
return {'version_count': self.count_versions()} return {'version_count': self.count_versions()}
return {}
def count_versions(self): def count_versions(self):
query = self.transaction_query() query = self.transaction_query()