Login with session in progress

Author: moithepro

app/src/main/java/com/moithepro/instatoolsandroid/LoginActivity.java

@@ -29,6 +29,8 @@
 import com.moithepro.instatoolsandroid.jInstaloader.JInvalidArgumentException;
 import com.moithepro.instatoolsandroid.jInstaloader.JTwoFactorAuthRequiredException;
 
+import java.io.FileNotFoundException;
+
 public class LoginActivity extends AppCompatActivity {
     private EditText password;
     private EditText username;
@@ -74,7 +76,11 @@ protected void onCreate(Bundle savedInstanceState) {
                             String u = username.getText().toString().trim();
                             String p = password.getText().toString().trim();
                             loader.login(u, p);
-
+                            try {
+                                loader.saveSession("session");
+                            } catch (Exception e) {
+                                e.printStackTrace();
+                            }
                             mainHandler.post(() -> {
                                 sp.edit().putString(getString(R.string.username_key), u).putString(getString(R.string.password_key), p).apply();
 
@@ -332,7 +338,40 @@ protected void onCreate(Bundle savedInstanceState) {
                 Handler mainHandler = new Handler(Looper.getMainLooper());
                 loginThread = new Thread(() -> {
 
+                    try {
+                        loader.loadSession("session");
+                        mainHandler.post(() -> {
+                            Intent intent = new Intent(this, UsersActivity.class);
+                            startActivity(intent);
+                            loggingIn = false;
+                            password.setEnabled(true);
+                            username.setEnabled(true);
+                            login.setText(R.string.log_in);
+                            finish();
+                            //setContentView(R.layout.users_activity);
+
+                        });
+                    }catch (FileNotFoundException e){
+                        mainHandler.post(() -> {
+                            AD = new AlertDialog.Builder(this)
+                                    .setTitle("Error")
+                                    .setMessage("Session File Not Found")
+
+                                    // Specifying a listener allows you to take an action before dismissing the dialog.
+                                    // The dialog is automatically dismissed when a dialog button is clicked.
+                                    .setPositiveButton(android.R.string.ok, null)
 
+                                    // A null listener allows the button to dismiss the dialog and take no further action.
+                                    .show();
+                            loggingIn = false;
+                            password.setEnabled(true);
+                            username.setEnabled(true);
+                            login.setText(R.string.log_in);
+                        });
+                    }
+                    catch (Exception e){
+                        e.printStackTrace();
+                    }
                     try {
                         loader.login(spu, spp);
 

app/src/main/java/com/moithepro/instatoolsandroid/jInstaloader/JInstaloader.java

@@ -4,6 +4,7 @@
 import com.chaquo.python.PyObject;
 import com.chaquo.python.Python;
 
+import java.io.FileNotFoundException;
 import java.io.Serializable;
 import java.util.ArrayList;
 import java.util.List;
@@ -18,6 +19,8 @@ public class JInstaloader {
     public static final int E_TwoFactorAuthRequiredException = 6;
     public static final int E_PrivateProfileNotFollowedException = 7;
     public static final int E_Exception = 8;
+
+    public static final int E_FileNotFoundError = 9;
     private Python py;
     private PyObject instaloaderInterface;
     private String loggedInUsername = null;
@@ -55,7 +58,30 @@ private void ignoreThisMethodItHasNoPracticalUsage(String username, String passw
                 throw new JException();
         }
     }
-
+    public void loadSession(String filename) throws FileNotFoundException, JException {
+        PyObject res = instaloaderInterface.callAttr("load_session", instaloaderInterface, filename);
+        int err = res.callAttr("get_e").toInt();
+        switch (err) {
+            case NO_ERROR:
+                break;
+            case E_FileNotFoundError:
+                throw new FileNotFoundException();
+            case E_Exception:
+                throw new JException();
+        }
+    }
+    public void saveSession(String filename) throws JException, JLoginRequiredException {
+        PyObject res = instaloaderInterface.callAttr("save_session", instaloaderInterface, filename);
+        int err = res.callAttr("get_e").toInt();
+        switch (err) {
+            case NO_ERROR:
+                break;
+            case E_LoginRequiredException:
+                throw new JLoginRequiredException();
+            case E_Exception:
+                throw new JException();
+        }
+    }
     public void login(String username, String password) throws JInvalidArgumentException, JConnectionException, JBadCredentialsException, JTwoFactorAuthRequiredException, JException {
 
         PyObject res = instaloaderInterface.callAttr("login", instaloaderInterface, username, password);

app/src/main/python/interface.py

@@ -14,7 +14,7 @@
 E_TwoFactorAuthRequiredException = 6
 E_PrivateProfileNotFollowedException = 7
 E_Exception = 8
-
+E_FileNotFoundError = 9
 
 class Result:
 
@@ -32,7 +32,27 @@ def get_arg(self):
 class Interface:
     def __init__(self):
         self.L: Instaloader = None
-
+    def load_session(self, session):
+        try:
+            print("load")
+            self.L = instaloader.Instaloader()
+            self.L.load_session_from_file(session)
+            return Result(NO_ERROR, 0)
+        except FileNotFoundError as err:
+            return Result(E_FileNotFoundError, 0)
+        except Exception as ex:
+            print(ex)
+            return Result(E_Exception, 0)
+    def save_session(self, session):
+        try:
+            print("save")
+            self.L.save_session_to_file(session)
+            return Result(NO_ERROR, 0)
+        except LoginRequiredException as err:
+            return Result(E_LoginRequiredException, 0)
+        except Exception as ex:
+            print(ex)
+            return Result(E_Exception, 0)
     def login(self, username, password):
         try:
             print("login")