import logging
import os

import pymysql as pymysql


def get_connection() -> pymysql.Connection:
	"""
		Get a connection to SQL.
		This function is used on the vServer for local testing
		@return: pymysql connection object
		"""
	try:
		if os.environ.get('IS_VSERVER') == 'true':
			conn = pymysql.connect(
				user=os.environ['vServer_SQL_User'],
				password=os.environ['vServer_SQL_Password'],
				host='localhost',
				port=3306,
				database='ka-news-yaa-crawler'
			)
		else:
			conn = pymysql.connect(
				user=os.environ['DB_USER'],
				password=os.environ['DB_PASSWORD'],
				host=os.environ['DB_HOST'],
				port=3306,
				database='ka-news-yaa-crawler'
			)

		return conn
	except pymysql.Error as e:
		logging.error('SQL Connection error: %s', e)
		return None