Skip to content

Commit

Permalink
[FIX] Missing import with per entity base class
Browse files Browse the repository at this point in the history
  • Loading branch information
tomguthrie committed May 8, 2012
1 parent 1a86e88 commit 8d04d24
Showing 1 changed file with 9 additions and 4 deletions.
13 changes: 9 additions & 4 deletions mogenerator.m
Original file line number Diff line number Diff line change
Expand Up @@ -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 {
Expand Down

0 comments on commit 8d04d24

Please sign in to comment.