fixed #96

tilltue · Mar 26, 2018 · 7360da1 · 7360da1 · sonjh1217 · Mar 27, 2018
1 parent b183f3b
commit 7360da1
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 5 deletions.
diff --git a/TLPhotoPicker.podspec b/TLPhotoPicker.podspec
@@ -8,7 +8,7 @@
 
 Pod::Spec.new do |s|
   s.name             = 'TLPhotoPicker'
-  s.version          = '1.5.4'
+  s.version          = '1.5.5'
   s.summary          = 'multiple phassets picker for iOS lib. like facebook'
 
 # This description is used to generate tags and improve search results.

diff --git a/TLPhotoPicker/Classes/TLPhotosPickerViewController.swift b/TLPhotoPicker/Classes/TLPhotosPickerViewController.swift
@@ -175,13 +175,17 @@ open class TLPhotosPickerViewController: UIViewController {
         stopPlay()
     }
 
-    override open func viewDidLoad() {
-        super.viewDidLoad()
-        makeUI()
+    func checkAuthorization() {
         switch PHPhotoLibrary.authorizationStatus() {
         case .notDetermined:
             PHPhotoLibrary.requestAuthorization { [weak self] status in
-                self?.initPhotoLibrary()
+                switch status {
+                case .authorized:
+                    self?.initPhotoLibrary()
+                default:
+                    self?.dismiss(done: false)
+                    self?.handleDeniedAlbumsAuthorization()
+                }
             }
         case .authorized:
             self.initPhotoLibrary()
@@ -191,6 +195,12 @@ open class TLPhotosPickerViewController: UIViewController {
         }
     }
 
+    override open func viewDidLoad() {
+        super.viewDidLoad()
+        makeUI()
+        checkAuthorization()
+    }
+
     override open func viewDidLayoutSubviews() {
         super.viewDidLayoutSubviews()
         if self.thumbnailSize == CGSize.zero {