Changeset 734 for branches/CoreData/SapphireFrappliance/MetaData/SapphireMObjects/SapphireMovie.m

Timestamp:

01/08/09 21:03:22 (3 years ago)

Author:

gbooker

Message:

Switched the the next version of the metadata.
Fixes #264

Files:

• branches/CoreData/SapphireFrappliance/MetaData/SapphireMObjects/SapphireMovie.m (modified) (6 diffs)

branches/CoreData/SapphireFrappliance/MetaData/SapphireMObjects/SapphireMovie.m

@@ -10 +10 @@
 #import "SapphireXMLData.h"
 #import "NSString-Extensions.h"
+#import "SapphireMovieTranslation.h"
 
 NSString *FILE_DID_CHANGE_MOVIE_NOTIFICATION = @"FileDidChangeMovie";
@@ -40 +41 @@
         ret = [NSEntityDescription insertNewObjectForEntityForName:SapphireMovieName inManagedObjectContext:moc];
         ret.imdbNumber = [NSNumber numberWithInt:imdbNumber];
+        return ret;
+}
+
++ (SapphireMovie *)movieWithTitle:(NSString *)title inContext:(NSManagedObjectContext *)moc
+{
+        NSPredicate *predicate = [NSPredicate predicateWithFormat:@"imdbNumber == nil && title == %@", title];
+        return (SapphireMovie *)doSingleFetchRequest(SapphireMovieName, moc, predicate);
+}
+
++ (SapphireMovie *)createMovieWithTitle:(NSString *)title inContext:(NSManagedObjectContext *)moc
+{
+        SapphireMovie *ret = [SapphireMovie movieWithTitle:title inContext:moc];
+        if(ret != nil)
+                return ret;
+        
+        ret = [NSEntityDescription insertNewObjectForEntityForName:SapphireMovieName inManagedObjectContext:moc];
+        ret.title = title;
+        ret.imdbNumber = nil;
         return ret;
 }
@@ -98 +117 @@
 }
 
++ (NSDictionary *)upgradeV1MoviesFromContext:(NSManagedObjectContext *)oldMoc toContext:(NSManagedObjectContext *)newMoc withCast:(NSDictionary *)cast directors:(NSDictionary *)directors genres:(NSDictionary *)genres
+{
+        NSMutableDictionary *lookup = [NSMutableDictionary dictionary];
+        NSArray *oldMovies = doFetchRequest(SapphireMovieName, oldMoc, nil);
+        NSEnumerator *movieEnum = [oldMovies objectEnumerator];
+        NSManagedObject *oldMovie;
+        while((oldMovie = [movieEnum nextObject]) != nil)
+        {
+                SapphireMovie *newMovie = [NSEntityDescription insertNewObjectForEntityForName:SapphireMovieName inManagedObjectContext:newMoc];
+                NSNumber *imdbNumber = [oldMovie valueForKey:@"imdbNumber"];
+                newMovie.imdbNumber = imdbNumber;
+                newMovie.imdbRating = [oldMovie valueForKey:@"imdbRating"];
+                newMovie.imdbTop250Ranking = [oldMovie valueForKey:@"imdbTop250Ranking"];
+                newMovie.MPAARating = [oldMovie valueForKey:@"MPAARating"];
+                newMovie.orderedCastData = [oldMovie valueForKey:@"orderedCastData"];
+                newMovie.orderedDirectorsData = [oldMovie valueForKey:@"orderedDirectorsData"];
+                newMovie.orderedGenresData = [oldMovie valueForKey:@"orderedGenresData"];
+                newMovie.oscarsWon = [oldMovie valueForKey:@"oscarsWon"];
+                newMovie.plot = [oldMovie valueForKey:@"plot"];
+                newMovie.releaseDate = [oldMovie valueForKey:@"releaseDate"];
+                newMovie.title = [oldMovie valueForKey:@"title"];
+                NSArray *castNames = [oldMovie valueForKeyPath:@"cast.name"];
+                NSEnumerator *castEnum = [castNames objectEnumerator];
+                NSString *name;
+                NSMutableSet *objSet = [NSMutableSet set];
+                while((name = [castEnum nextObject]) != nil)
+                {
+                        [objSet addObject:[cast objectForKey:name]];
+                }
+                [newMovie.castSet setSet:objSet];
+                NSArray *directorNames = [oldMovie valueForKeyPath:@"directors.name"];
+                NSEnumerator *directorEnum = [directorNames objectEnumerator];
+                objSet = [NSMutableSet set];
+                while((name = [directorEnum nextObject]) != nil)
+                {
+                        [objSet addObject:[directors objectForKey:name]];
+                }
+                [newMovie.directorsSet setSet:objSet];
+                NSArray *genreNames = [oldMovie valueForKeyPath:@"genres.name"];
+                NSEnumerator *genreEnum = [genreNames objectEnumerator];
+                objSet = [NSMutableSet set];
+                while((name = [genreEnum nextObject]) != nil)
+                {
+                        [objSet addObject:[genres objectForKey:name]];
+                }
+                [newMovie.genresSet setSet:objSet];
+                
+                NSEnumerator *translationEnum = [[oldMovie valueForKey:@"translations"] objectEnumerator];
+                NSManagedObject *translation;
+                while((translation = [translationEnum nextObject]) != nil)
+                {
+                        [SapphireMovieTranslation upgradeV1MovieTranslation:translation toMovie:newMovie];
+                }
+                [lookup setObject:newMovie forKey:imdbNumber];
+        }
+        return lookup;
+}
+
 - (NSComparisonResult)releaseDateCompare:(SapphireMovie *)other
 {
@@ -275 +352 @@
         checkOrderedData(orderedCast, overridden);
         if(![overridden count])
-        {
                 self.overriddenCastData = nil;
+        else
+        {
+                NSArray *names = [overridden valueForKey:@"name"];
+                self.overriddenCastData = [NSKeyedArchiver archivedDataWithRootObject:names];
         }
         [self setPrimitiveValue:nil forKey:ORDERED_CAST_KEY];
@@ -287 +367 @@
         checkOrderedData(orderedGenres, overridden);
         if(![overridden count])
-        {
                 self.overriddenGenresData = nil;
+        else
+        {
+                NSArray *names = [overridden valueForKey:@"name"];
+                self.overriddenGenresData = [NSKeyedArchiver archivedDataWithRootObject:names];
         }
         [self setPrimitiveValue:nil forKey:ORDERED_GENRES_KEY];
@@ -299 +382 @@
         checkOrderedData(orderedDirectors, overridden);
         if(![overridden count])
-        {
                 self.overriddenDirectorsData = nil;
+        else
+        {
+                NSArray *names = [overridden valueForKey:@"name"];
+                self.overriddenDirectorsData = [NSKeyedArchiver archivedDataWithRootObject:names];
         }
         [self setPrimitiveValue:nil forKey:ORDERED_DIRECTOR_KEY];