1 files changed, 13 insertions, 0 deletions
diff --git a/src/com/android/gallery3d/ui/ImportCompleteListener.java b/src/com/android/gallery3d/ui/ImportCompleteListener.java
index 63d616751..5b6e0a17f 100644
--- a/src/com/android/gallery3d/ui/ImportCompleteListener.java
+++ b/src/com/android/gallery3d/ui/ImportCompleteListener.java
@@ -16,7 +16,10 @@
 
 package com.android.gallery3d.ui;
 
+import android.app.Activity;
+import android.content.Context;
 import android.os.Bundle;
+import android.os.PowerManager;
 import android.widget.Toast;
 
 import com.android.gallery3d.R;
@@ -26,9 +29,13 @@ import com.android.gallery3d.util.MediaSetUtils;
 
 public class ImportCompleteListener implements MenuExecutor.ProgressListener {
     private GalleryActivity mActivity;
+    private PowerManager.WakeLock mWakeLock;
 
     public ImportCompleteListener(GalleryActivity galleryActivity) {
         mActivity = galleryActivity;
+        PowerManager pm =
+                (PowerManager) ((Activity) mActivity).getSystemService(Context.POWER_SERVICE);
+        mWakeLock = pm.newWakeLock(PowerManager.SCREEN_DIM_WAKE_LOCK, "Gallery Album Import");
     }
 
     @Override
@@ -41,12 +48,18 @@ public class ImportCompleteListener implements MenuExecutor.ProgressListener {
             message = R.string.import_fail;
         }
         Toast.makeText(mActivity.getAndroidContext(), message, Toast.LENGTH_LONG).show();
+        mWakeLock.release();
     }
 
     @Override
     public void onProgressUpdate(int index) {
     }
 
+    @Override
+    public void onProgressStart() {
+        mWakeLock.acquire();
+    }
+
     private void goToImportedAlbum() {
         String pathOfImportedAlbum = "/local/all/" + MediaSetUtils.IMPORTED_BUCKET_ID;
         Bundle data = new Bundle();