Merge pull request #350 from ChristopherRogers/master

[CHANGE] Swift 3: NSData to Data. ([Christopher Rogers](#350))
rentzsch · Oct 10, 2016 · 64dbbbb · 64dbbbb
2 parents 1c77f4e + 1992cdd
commit 64dbbbb
Showing 1 changed file with 8 additions and 5 deletions.
diff --git a/mogenerator.m b/mogenerator.m
@@ -550,11 +550,14 @@ - (NSString*)objectAttributeClassName {
     } else {
         result = [self attributeValueClassName];
     }
-    if (gSwift && [result isEqualToString:@"NSString"]) {
-        result = @"String";
-    }
-    if (gSwift && [result isEqualToString:@"NSDate"]) {
-        result = @"Date";
+    if (gSwift) {
+        if ([result isEqualToString:@"NSString"]) {
+            result = @"String";
+        } else if ([result isEqualToString:@"NSDate"]) {
+            result = @"Date";
+        } else if ([result isEqualToString:@"NSData"]) {
+            result = @"Data";
+        }
     }
     return result;
 }