Handle case where +postWithAttributes: returns nil.

isutton · Jul 14, 2010 · f51c3b5 · f51c3b5
1 parent 8823213
commit f51c3b5
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/TKPost.m b/TKPost.m
@@ -113,7 +113,7 @@ - (NSString *)description
 
 + (id)postWithAttributes:(NSDictionary *)attributeDict
 {
-    Class postClass;
+    Class postClass = nil;
     NSString *type_ = [attributeDict objectForKey:@"type"];
 
     for (int i = 0; TKPostTypeStringToClassName[i].name != nil; i++) {

diff --git a/TKTumblrReader.m b/TKTumblrReader.m
@@ -59,7 +59,8 @@ - (void)parser:(NSXMLParser *)parser didStartElement:(NSString *)elementName nam
 - (void)parser:(NSXMLParser *)parser didEndElement:(NSString *)elementName namespaceURI:(NSString *)namespaceURI qualifiedName:(NSString *)qName
 {
     if ([elementName isEqualToString:@"post"]) {
-        [posts addObject:currentPost];
+        if (currentPost != nil)
+            [posts addObject:currentPost];
         self.currentPost = nil;
     }
 }