services.py

from selenium import webdriver
from selenium.webdriver.chrome.options import Options
from bs4 import BeautifulSoup
from termcolor import colored
from sys import platform
from pyfiglet import figlet_format
import requests
import os
import random
import click

#######################
###### STACK CLI ######
#######################

#Defining global parameters to reduce complexity
#System variables to ensure proper functioning
global chrome_options
global chromedriver
global driver
global questions
global answers

print(figlet_format('StackCLI', font='starwars'))

def configure_headless():
	global chrome_options
	global chromedriver
	global driver

	# Initiang the browser in headless mode
	# --headless option can be removed to view the browser
	# Using 1920x1080 as default resolution
	
	chrome_options = Options()
	chrome_options.add_argument("--headless")
	chrome_options.add_argument("--window-size=1920x1080")

	# Setting webdriver files based on the platform StackCLI is running
	
	if platform == "linux" or platform == "linux2":
	    name = "/linux"
	elif platform == "darwin":
	    name = "/mac"
	elif platform == "win32":
	    name = "/win.exe"
	#Platform detection enabled for chromedriver files
	chrome_driver = os.getcwd() + name
	#Passing location of the chromedriver executable
	driver = webdriver.Chrome(chrome_options=chrome_options, executable_path=chrome_driver)

def nth_answer(n, question, option=''):
	global questions
	
	#Passing no parameter to ensure, print is not initiated
	
	get_answers(question, 'NULL')
	link = questions[n]
	
	# Searching for links (Iterating through the <a href></a> tags
	
	for a in link.find_all('a'):
		if a.has_attr('href'):
			ans = a.attrs['href']

	# If user request for Nth answer, newurl needs to be generated
	
	url = 'http://stackoverflow.com'+ans 
	if option == 'screenshot':
		driver.get(url)
		filename = str(random.randint(0,1000))
		driver.get_screenshot_as_file(filename + '.png')
		
		# Getting a random name for screenshot
		# ensuring previous ones don't get overwritten

	else:
		page = requests.get(url)
		data = page.text
		soup = BeautifulSoup(data,'html.parser')

		soup = soup.find('div', {"id":"content"})

		heading = soup.find('a', {"class":"question-hyperlink"})
		click.echo(colored('Q : '+heading.text, 'red'))

		body = soup.find('div', {"class":"post-text"})
		click.echo(colored(body.text, 'blue'))
		
		# Getting content from the page



def stackoverflow(question):
	configure_headless()
	# Setting option by calling configure_headless()
	
	global chrome_options
	global chromedriver
	global driver

	#opening stackoverflow in the headless chrome browser
	driver.get("https://stackoverflow.com/")

	#location search box using the Xpath (XML Path)
	search = driver.find_element_by_name("q")
	search.send_keys(question)
	#writing queries to the search box using send_keys
	search_button = driver.find_element_by_xpath("/html/body/header/div/form/div/button")
	search_button.click()
	#Simulating click on the seach button on the page

	return driver.current_url
	#Returning the url of the page after initiating click

def display_answers(questions, answers):
	index = 0
	#Picking elements from the question and answers list using the index
	for index in range(0, len(questions)):
		#Priting the question on index
		click.echo(colored(questions[index].text.replace("  ",""), 'red'))

		link = questions[index]
		for a in link.find_all('a'):
			if a.has_attr('href'):
				ans = a.attrs['href']

		#Generating clickable link on the terminal
		click.echo(colored('http://stackoverflow.com'+ans, 'green'))
		#Priting the answer on the index
		click.echo(colored(answers[index].text.replace("  ",""), 'blue'))

def kill_browser():
	#killing the webdriver processes for cleaning up the memory due to
	#unexpected network error
	
	driver.close()
	driver.quit()

def get_answers(question, option=''):
	global questions
	url = stackoverflow(question)
	page = requests.get(url)
	data = page.text
	soup = BeautifulSoup(data,'html.parser')
	#generating the source of the page and locating Q&A section
	soup = soup.find("div", {"class":"js-search-results"})


	try:
		questions = soup.find_all('div', {"class":"result-link"})
		answers = soup.find_all('div', {"class":"excerpt"})
		#eaching for questions and answers using the class id's
		if option == "search":
			display_answers(questions, answers)

	except:
		#get_answers(question, option)
		click.echo('Network Error --> Killing Browser')

def main(question, screenshot):
	global driver
	get_answers(question, 'screenshot')
	filename = str(random.randint(0,1000))
	driver.get_screenshot_as_file(filename + '.png')
	#Generating screenshot and saving it with a random name