Merge pull request rentzsch#109 from tguthrie/mogenerator

--- Wasnt importing any header when using `mogenerator.customBaseClass` entity userInfo key as `-[NSEntityDescription hasCustomSuperentity]` wasnt aware of the option.
ddrccw · May 10, 2012 · b9fd6aa · b9fd6aa
2 parents f671c53 + 5d165f7
commit b9fd6aa
Showing 1 changed file with 9 additions and 4 deletions.
diff --git a/mogenerator.m b/mogenerator.m
@@ -73,11 +73,16 @@ - (NSArray*)entitiesWithACustomSubclassInConfiguration:(NSString *)configuration
 
 @implementation NSEntityDescription (customBaseClass)
 - (BOOL)hasCustomSuperentity {
-	NSEntityDescription *superentity = [self superentity];
-	if (superentity) {
-		return YES;
+	NSString *forcedBaseClass = [self forcedCustomBaseClass];
+	if (!forcedBaseClass) {
+		NSEntityDescription *superentity = [self superentity];
+		if (superentity) {
+			return YES;
+		} else {
+			return gCustomBaseClass ? YES : NO;
+		}
 	} else {
-		return gCustomBaseClass ? YES : NO;
+		return YES;
 	}
 }
 - (NSString*)customSuperentity {