def socli(query): """ SoCLI Code :param query: Query to search on Stack Overflow. If google_search is true uses Google search to find the best result. Else, use Stack Overflow default search mechanism. :return: """ query = printer.urlencode(query) try: if search.google_search: questions = search.get_questions_for_query_google(query) res_url = questions[0][2] # Gets the first result display_results(res_url) else: questions = search.get_questions_for_query(query) res_url = questions[0][2] display_results(search.so_url + res_url) # Returned URL is relative to SO homepage except IndexError: if len(questions) > 1: for i in range(1, len(questions)): res_url = questions[i][2] try: display_results(res_url) except IndexError: continue break printer.print_warning("No results found...") except UnicodeEncodeError as e: printer.showerror(e) printer.print_warning("\n\nEncoding error: Use \"chcp 65001\" command before using socli...") sys.exit(0) except requests.exceptions.ConnectionError: printer.print_fail("Please check your internet connectivity...") except Exception as e: printer.showerror(e) sys.exit(0)
def main(): """ The main logic for how options in a command is checked. """ global query namespace = parse_arguments(sys.argv[1:]) search.load_user_agents() # Populates the user agents array query_tag = ' '.join(namespace.browse) # Tags # Query if namespace.query: # Query when args are present query = ' '.join(namespace.query) elif namespace.userQuery: # Query without any args query = ' '.join(namespace.userQuery) if namespace.help: # Display command line syntax printer.helpman() sys.exit(0) if namespace.debug: # If --debug flag is present # Prints out error used for debugging printer.DEBUG = True debug_requests_on() if namespace.new: # If --new flag is present # Opens StackOverflow website in the browser to create a new question import webbrowser printer.print_warning("Opening stack overflow in your browser...") webbrowser.open(search.so_url + "/questions/ask") sys.exit(0) if namespace.api: # If --api flag is present # Sets custom API key user_module.set_api_key() sys.exit(0) if namespace.user is not None: # If --user flag is present # Stackoverflow user profile support if namespace.user != '(RANDOM_STRING_CONSTANT)': # If user provided a user ID user_module.manual = 1 # Enabling manual mode user = namespace.user else: # If user did not provide a user id user = user_module.retrieve_saved_profile() # Reading saved user id from app data user_module.user_page(user) sys.exit(0) if namespace.delete: # If --delete flag is present # Deletes user data user_module.del_datafile() printer.print_warning("Data files deleted...") sys.exit(0) if namespace.sosearch: # If --sosearch flag is present # Disables google search search.google_search = False if namespace.tag: # If --tag flag is present global tag search.google_search = False tag = namespace.tag has_tags() # Adds tags to StackOverflow url (when not using google search. if namespace.open_url: import webbrowser open_in_browser=False display_condition=True url_to_use=namespace.open_url[0] if re.findall(r"^https:\/\/",url_to_use) !=[]: pass else: url_to_use="https://" + url_to_use try: if url_to_use == "https://stackoverflow.com/questions/": raise Exception('URL Error') if url_to_use == "https://www.stackoverflow.com/questions/": raise Exception('URL Error') requests.get(url_to_use) except Exception: printer.print_warning("Error, could be:\n- invalid url\n- url cannot be opened in socli\n- internet connection error") sys.exit(0) nostackoverflow=re.findall(r"stackoverflow\.com",url_to_use) if nostackoverflow == []: open_in_browser=True display_condition=False printer.print_warning("Your url is not a stack overflow url.\nOpening in your browser...") tag_matcher=re.findall(r"\/tag.+\/",url_to_use) blog_matcher=re.findall(r"blog",url_to_use) if tag_matcher != []: extracted_tag="" if re.findall(r"tagged",url_to_use) == []: extracted_tag=re.split(r"\/",url_to_use)[4] else: extracted_tag=re.split(r"\/",url_to_use)[5] open_in_browser=False display_condition=False tag=extracted_tag search.socli_interactive(tag) if blog_matcher != []: open_in_browser=True display_condition=False printer.print_warning("Your url belongs to blog") printer.print_warning("Opening in browser...") if display_condition: open_in_browser=False display_results(url_to_use) if open_in_browser: webbrowser.open(url_to_use) if namespace.res is not None: # If --res flag is present # Automatically displays the result specified by the number question_number = namespace.res if namespace.query != [] or namespace.tag is not None: # There must either be a tag or a query search.socli_manual_search(query, question_number) else: printer.print_warning('You must specify a query or a tag. For example, use: "socli -r 3 -q python for loop" ' 'to retrieve the third result when searching about "python for loop". ' 'You can also use "socli -r 3 -t python" ' 'to retrieve the third result when searching for posts with the "python" tag.') if namespace.browse: # Browse mode search.google_search = False socli_browse_interactive(query_tag) elif namespace.query != [] or namespace.tag is not None: # If query and tag are not both empty if namespace.interactive: search.socli_interactive(query) else: socli(query) elif query not in [' ', ''] and not ( namespace.tag or namespace.res or namespace.interactive or namespace.browse): # If there are no flags socli(query) else: # Help text for interactive mode if namespace.interactive and namespace.query == [] and namespace.tag is None: printer.print_warning('You must specify a query or a tag. For example, use: "socli -iq python for loop" ' 'to search about "python for loop" in interactive mode. ' 'You can also use "socli -it python" ' 'to search posts with the "python" tag in interactive mode.') else: printer.helpman()
def socli_browse_interactive_windows(query_tag): """ Interactive mode for -b browse :param query_tag: :return: """ try: search_res = requests.get(search.so_burl + query_tag) search.captcha_check(search_res.url) soup = BeautifulSoup(search_res.text, 'html.parser') try: soup.find_all("div", class_="question-summary")[0] # For explicitly raising exception tmp = (soup.find_all("div", class_="question-summary")) i = 0 question_local_url = [] print(printer.bold("\nSelect a question below:\n")) while i < len(tmp): if i == 10: break # limiting results question_text = ' '.join((tmp[i].a.get_text()).split()) question_text = question_text.replace("Q: ", "") printer.print_warning(str(i + 1) + ". " + printer.display_str(question_text)) q_tag = (soup.find_all("div", class_="question-summary"))[i] answers = [s.get_text() for s in q_tag.find_all("a", class_="post-tag")][0:] ques_tags = " ".join(str(x) for x in answers) question_local_url.append(tmp[i].a.get("href")) print(" " + printer.display_str(ques_tags) + "\n") i = i + 1 try: op = int(printer.inputs("\nType the option no to continue or any other key to exit:")) while 1: if (op > 0) and (op <= i): display_results(search.so_burl + question_local_url[op - 1]) cnt = 1 # this is because the 1st post is the question itself while 1: global tmpsoup qna = printer.inputs( "Type " + printer.bold("o") + " to open in browser, " + printer.bold( "n") + " to next answer, " + printer.bold( "b") + " for previous answer or any other key to exit:") if qna in ["n", "N"]: try: answer = (tmpsoup.find_all("div", class_="js-post-body")[cnt + 1].get_text()) printer.print_green("\n\nAnswer:\n") print("-------\n" + answer + "\n-------\n") cnt = cnt + 1 except IndexError: printer.print_warning(" No more answers found for this question. Exiting...") sys.exit(0) continue elif qna in ["b", "B"]: if cnt == 1: printer.print_warning(" You cant go further back. You are on the first answer!") continue answer = (tmpsoup.find_all("div", class_="js-post-body")[cnt - 1].get_text()) printer.print_green("\n\nAnswer:\n") print("-------\n" + answer + "\n-------\n") cnt = cnt - 1 continue elif qna in ["o", "O"]: import webbrowser printer.print_warning("Opening in your browser...") webbrowser.open(search.so_burl + question_local_url[op - 1]) else: break sys.exit(0) else: op = int(input("\n\nWrong option. select the option no to continue:")) except Exception as e: printer.showerror(e) printer.print_warning("\n Exiting...") sys.exit(0) except IndexError: printer.print_warning("No results found...") sys.exit(0) except UnicodeEncodeError: printer.print_warning("\n\nEncoding error: Use \"chcp 65001\" command before using socli...") sys.exit(0) except requests.exceptions.ConnectionError: printer.print_fail("Please check your internet connectivity...") except Exception as e: printer.showerror(e) sys.exit(0)