search.py

#!bin/python3
import os
import json
from functions import duckduckgo
from functions import bing
from functions import facebook
import re
import urllib.request as urllib2

#This is a hardcoded query that will be a user input, the inurl & query tuning should be easier options to find or implicitly added
rawSearchQuery = "free clinic massachusetts"
inurl = ["contact"]
exclude = ["animals"]
#transform the query into a usable string
searchQuery = rawSearchQuery.replace(" ","+")
rpattern = "(\d{3}[-\.\s]??\d{3}[-\.\s]??\d{4}|\(\d{3}\)\s*\d{3}[-\.\s]??\d{4}|\d{3}[-\.\s]??\d{4})"
missingMessage = "No data found, search page directly"
#pass queries to search engines

#This is to transform ddg results from the form of [{url:"url",answer:"answer"}]
def getDDG(query,inurl=[],exclude=[]):
	inurltext = ""
	excludeText = ""
	for url in inurl:
		inurltext = inurltext+"+inurl:"+url
	for exc in exclude:
		excludeText = excludeText+"+-"+exc
	query = query+inurltext+excludeText
	ddgResults = duckduckgo.query(query)
	for result in ddgResults:
		phoneNum = missingMessage
		#parse out phone numbers
		phoneNumParse = re.search(rpattern, result["answer"])
		if phoneNumParse:
			phoneNum = phoneNumParse.group(1)
		#This outputs to the console, but it should respond with something more useful
		print("DuckDuckGo\n"+str(urllib2.unquote(result["url"]))+"\n"+result["description"]+"\n"+result["details"]+"\n"+phoneNum+"\n-------------\n")

def getBing(query,inurl=[],exclude=[]):
	inurltext = ""
	excludeText = ""
	for url in inurl:
		inurltext = inurltext+"+intitle:"+url
	for exc in exclude:
		excludeText = excludeText+"+-"+exc
	query = query+inurltext+excludeText
	bingResults = bing.query(query)
	for result in bingResults:
		phoneNum = missingMessage
		phoneNumParse = re.search(rpattern, result["answer"])
		if phoneNumParse:
			phoneNum = phoneNumParse.group(1)
		print("Bing\n"+str(urllib2.unquote(result["url"]))+"\n"+result["description"]+"\n"+result["details"]+"\n"+phoneNum+"\n-------------\n")
	return

def getFacebook():
	facebookResults = facebook.query()
	for result in facebookResults["data"]:
		phoneNum = missingMessage
		description = missingMessage
		about = missingMessage
		try:
			if result["phone"]:
				phoneNum = result["phone"]
			if result["description"]:
				description = result["description"]
			if result["about"]:
				about = result["about"]
		except:
			continue
		print("Facebook\n"+result["link"]+"\n"+result["about"]+"\n"+description+"\n"+phoneNum+"\n---------------\n")

#add or remove desired search queries here:
getDDG(searchQuery,inurl,exclude)
getBing(searchQuery,inurl,exclude)
getFacebook()