Skip to content

Commit 6b6b987

Browse files
committed
更新支持swift模型类生成
1 parent 8ad6f3d commit 6b6b987

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

WHC_DataModelFactory/ViewController.m

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@
2323
#define kWHC_CLASS_M @("@implementation %@\n\n@end\n")
2424

2525

26-
#define kSWHC_CLASS @("\nclass %@ :NSObject{\n%@\n}")
26+
#define kSWHC_CLASS @("\n@objc(%@)\nclass %@ :NSObject{\n%@\n}")
2727
#define kSWHC_PROPERTY @("var %@: %@!;\n")
2828
@interface ViewController (){
2929
NSMutableString * _classString; //存类头文件内容
@@ -80,7 +80,7 @@ - (IBAction)clickMakeButton:(NSButton*)sender{
8080
[_classMString appendFormat:kWHC_CLASS_M,className];
8181
[_classString appendFormat:kWHC_CLASS,className,[self handleDataEngine:dict key:@""]];
8282
}else{
83-
[_classString appendFormat:kSWHC_CLASS,className,[self handleDataEngine:dict key:@""]];
83+
[_classString appendFormat:kSWHC_CLASS,className,className,[self handleDataEngine:dict key:@""]];
8484
}
8585
_classField.stringValue = _classString;
8686
_classMField.stringValue = _classMString;
@@ -110,7 +110,7 @@ - (NSString*)handleDataEngine:(id)object key:(NSString*)key{
110110
[_classMString appendFormat:kWHC_CLASS_M,keyArr[i]];
111111
}else{
112112
[property appendFormat:kSWHC_PROPERTY,keyArr[i],keyArr[i]];
113-
[_classString appendFormat:kSWHC_CLASS,keyArr[i],classContent];
113+
[_classString appendFormat:kSWHC_CLASS,keyArr[i],keyArr[i],classContent];
114114
}
115115
}else if ([subObject isKindOfClass:[NSArray class]]){
116116
NSString * classContent = [self handleDataEngine:subObject key:keyArr[i]];
@@ -120,7 +120,7 @@ - (NSString*)handleDataEngine:(id)object key:(NSString*)key{
120120
[_classMString appendFormat:kWHC_CLASS_M,keyArr[i]];
121121
}else{
122122
[property appendFormat:kSWHC_PROPERTY,keyArr[i],@"NSArray"];
123-
[_classString appendFormat:kSWHC_CLASS,keyArr[i],classContent];
123+
[_classString appendFormat:kSWHC_CLASS,keyArr[i],keyArr[i],classContent];
124124
}
125125
}else if ([subObject isKindOfClass:[NSString class]]){
126126
if(_checkBox.state == 0){

0 commit comments

Comments
 (0)