PyInstaLive/pyinstalive/auth.py

import codecs
import datetime
import json
import os.path
import sys

from .logger import log_seperator, supports_color, log_info_blue, log_info_green, log_warn, log_error, log_whiteline, log_plain




try:
	from instagram_private_api import (
		Client, ClientError, ClientLoginError,
		ClientCookieExpiredError, ClientLoginRequiredError)
except ImportError:
	import sys
	sys.path.append(os.path.join(os.path.dirname(__file__), '..'))
	from instagram_private_api import (
		Client, ClientError, ClientLoginError,
		ClientCookieExpiredError, ClientLoginRequiredError)


def to_json(python_object):
	if isinstance(python_object, bytes):
		return {'__class__': 'bytes',
				'__value__': codecs.encode(python_object, 'base64').decode()}
	raise TypeError(repr(python_object) + ' is not JSON serializable')


def from_json(json_object):
	if '__class__' in json_object and json_object.get('__class__') == 'bytes':
		return codecs.decode(json_object.get('__value__').encode(), 'base64')
	return json_object


def onlogin_callback(api, cookie_file):
	cache_settings = api.settings
	with open(cookie_file, 'w') as outfile:
		json.dump(cache_settings, outfile, default=to_json)
		log_info_green('New cookie file was made: {0!s}'.format(cookie_file))
		log_seperator()


def login(username, password, show_cookie_expiry, force_use_login_args):
	device_id = None
	try:
		if force_use_login_args:
			log_info_blue("Overriding configuration file login with -u and -p arguments...")
			log_seperator()
		cookie_file = "{}.json".format(username)
		if not os.path.isfile(cookie_file):
			# settings file does not exist
			log_warn('Unable to find cookie file: {0!s}'.format(cookie_file))
			log_info_green('Creating a new cookie file...')

			# login new
			api = Client(
				username, password,
				on_login=lambda x: onlogin_callback(x, cookie_file))
		else:
			with open(cookie_file) as file_data:
				cached_settings = json.load(file_data, object_hook=from_json)
			# log_info_green('Using settings file: {0!s}'.format(cookie_file))

			device_id = cached_settings.get('device_id')
			# reuse auth cached_settings
			api = Client(
				username, password,
				settings=cached_settings)

	except (ClientCookieExpiredError) as e:
		log_warn('The current cookie file for "{:s}" has expired, creating a new one...'.format(username))

		# Login expired
		# Do relogin but use default ua, keys and such
		try:
			api = Client(
				username, password,
				device_id=device_id,
				on_login=lambda x: onlogin_callback(x, cookie_file))
		except Exception as ee:
			log_seperator()
			log_error('An error occurred while trying to create a new cookie file: {:s}'.format(str(ee)))
			if "getaddrinfo failed" in str(ee):
				log_error('Could not resolve host, check your internet connection.')
			elif "timed out" in str(ee):
				log_error('The connection timed out, check your internet connection.')
			elif "bad_password" in str(ee):
				log_error('The password you entered is incorrect. Please try again.')
			else:
				log_error('{:s}'.format(ee.message))
			log_seperator()
			exit(1)

	except ClientLoginError as e:
		log_seperator()
		log_error('Could not login: {:s}.'.format(json.loads(e.error_response).get("error_title", "Error title not available.")))
		log_error('{:s}'.format(json.loads(e.error_response).get("message", "Not available")))
		log_error('{:s}'.format(e.error_response))
		log_seperator()
		sys.exit(9)
	except ClientError as e:
		log_seperator()
		try:
			log_error('Unexpected exception: {0!s}'.format(e.msg))
			log_error('Message: {1!s}'.format(json.loads(e.error_response).get("message", "Additional error information not available.")))
			log_error('Code: {2:d}'.format(e.code))
			log_error('Full response:\n{3!s}'.format(e.error_response))
			log_whiteline()
		except Exception as ee:
			log_error('An error occurred while trying to handle a previous exception.')
			log_error('1: {:s}'.format(str(e)))
			log_error('2: {:s}'.format(str(ee)))
			if "getaddrinfo failed" in str(ee):
				log_error('Could not resolve host, check your internet connection.')
			if "timed out" in str(ee):
				log_error('The connection timed out, check your internet connection.')
		log_seperator()
		sys.exit(9)
	except Exception as e:
		if (str(e).startswith("unsupported pickle protocol")):
			log_warn("This cookie file is not compatible with Python {}.".format(sys.version.split(' ')[0][0]))
			log_warn("Please delete your cookie file '{}.json' and try again.".format(username))
		else:
			log_seperator()
			log_error('Unexpected exception: {0!s}'.format(e))
		log_seperator()
		sys.exit(99)
	except KeyboardInterrupt as e:
		log_seperator()
		log_warn("The user authentication has been aborted.")
		log_seperator()
		sys.exit(0)

	log_info_green('Successfully logged into user "{:s}".'.format(str(api.authenticated_user_name)))
	if show_cookie_expiry.title() == 'True' and not force_use_login_args:
		try:
			cookie_expiry = api.cookie_jar.auth_expires
			log_info_green('Cookie file expiry date: {0!s}'.format(datetime.datetime.fromtimestamp(cookie_expiry).strftime('%Y-%m-%d at %I:%M:%S %p')))
		except AttributeError as e:
			log_warn('An error occurred while getting the cookie file expiry date: {0!s}'.format(e))

	log_seperator()		
	return api