修复 bundle 加载问题

ksvc · Mar 22, 2018 · 582456a · 582456a
1 parent 19246dc
commit 582456a
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/PhotoPickerKit/KSYAlbumViewController.m b/PhotoPickerKit/KSYAlbumViewController.m
@@ -54,7 +54,8 @@ - (void)configSubviews{
                     strongSelf.tableView.tableFooterView = [[UIView alloc] init];
                     strongSelf.tableView.dataSource = self;
                     strongSelf.tableView.delegate = self;
-                    [strongSelf.tableView registerNib:[UINib nibWithNibName:@"KSYAlbumCell" bundle:nil] forCellReuseIdentifier:@"KSYAlbumCell"];
+                    NSBundle *bundle = [NSBundle bundleForClass:[KSYPhotoPickerController class]];
+                    [strongSelf.tableView registerNib:[UINib nibWithNibName:@"KSYAlbumCell" bundle:bundle] forCellReuseIdentifier:@"KSYAlbumCell"];
                     [strongSelf.view addSubview:strongSelf.tableView];
                     [strongSelf.tableView mas_makeConstraints:^(MASConstraintMaker *make) {
                         make.edges.equalTo(strongSelf.view);
@@ -121,7 +122,7 @@ - (CGFloat)tableView:(UITableView *)tableView heightForRowAtIndexPath:(NSIndexPa
 - (void)tableView:(UITableView *)tableView didSelectRowAtIndexPath:(NSIndexPath *)indexPath{
     KSYAlbumModel *model = self.albums[indexPath.row];
 
-    KSYAssetViewController *assetVC = [[KSYAssetViewController alloc] initWithNibName:@"KSYAssetViewController" bundle:nil];
+    KSYAssetViewController *assetVC = [[KSYAssetViewController alloc] initWithNibName:@"KSYAssetViewController" bundle:[NSBundle bundleForClass:[KSYPhotoPickerController class]]];
     assetVC.model = model;
     [self.navigationController pushViewController:assetVC animated:YES];
 }

diff --git a/PhotoPickerKit/KSYAssetViewController.m b/PhotoPickerKit/KSYAssetViewController.m
@@ -111,7 +111,8 @@ - (void)configCollectionView {
     self.collectionView.contentInset = UIEdgeInsetsMake(itemMargin, itemMargin, itemMargin, itemMargin);
 
     self.collectionView.contentSize = CGSizeMake(CGRectGetWidth(self.view.frame), ((self.model.assetsCount + 4 - 1) / 4) * CGRectGetWidth(self.view.frame));
-    UINib *nib = [UINib nibWithNibName:@"KSYAssetCell" bundle:[NSBundle mainBundle]];
+    NSBundle *bundle = [NSBundle bundleForClass:[KSYPhotoPickerController class]];
+    UINib *nib = [UINib nibWithNibName:@"KSYAssetCell" bundle:bundle];
     [self.collectionView registerNib:nib forCellWithReuseIdentifier:@"KSYAssetCell"];
 }