X-Git-Url: http://git.onelab.eu/?a=blobdiff_plain;f=planetstack%2Fapigen%2Fmodelgen;fp=planetstack%2Fapigen%2Fmodelgen;h=e0def4b356803a6fc4a823903eada54001422447;hb=df2b49ed09b8a3f6734f175ad9d41c953be5bed9;hp=d2e80f53871ee91ff141ab999561c90979dc7a60;hpb=643b70979a1baf6b9da44560768cd35c5aa9e435;p=plstackapi.git

diff --git a/planetstack/apigen/modelgen b/planetstack/apigen/modelgen
index d2e80f5..e0def4b 100755
--- a/planetstack/apigen/modelgen
+++ b/planetstack/apigen/modelgen
@@ -7,6 +7,8 @@ import json
 import re
 from django.template import Context, Template
 
+blacklist = ['SingletonModel','PlCoreBase']
+
 # Django set up
 
 sys.path.append('.')
@@ -27,7 +29,7 @@ g = globals()
 def enum_classes():
 	model_classes = []
 	for c in g.values():
-		if type(c)==type(PlCoreBase):
+		if type(c)==type(PlCoreBase) and c.__name__ not in blacklist:
 			model_classes.append(c)
 	return model_classes
 
@@ -50,7 +52,7 @@ class GenObj(object):
 
 	def camel(self):
 		name = str(self.model.__name__)
-		return name.title()
+		return name
 		
 class Generator(dict):
 	def all(self):
@@ -67,6 +69,8 @@ class Generator(dict):
 
 	def compute_links(self):
 		for obj in self.values():
+			if (str(obj).lower()=='user'):
+				pdb.set_trace()
 			fields = obj.model._meta.fields
 			for f in fields:
 				if type(f)==ForeignKey and f.name in self.keys():