def validateInstaUser(username, num_jitters):
images = getInstaLinks(username)
if len(images) >= cfg.instaLimit():
images = images[:cfg.instaLimit()]
r = FaceRecog(username, images, num_jitters=num_jitters)
r.loadKnown(username)
profile_links, _ = r.getValidLinksAndImg(username)
return len(profile_links) > 0
def getInstaLinks(username):
looter = InstaLooter(profile=username)
images = []
i = 0
for media in looter.medias():
if i > cfg.instaLimit():
break
if not media['is_video']:
console.subtask("Got Image: {0}".format(media['display_src'].strip()[:90]))
images.append(media['display_src'])
i = i + 1
return images
def main(skipFB=False, skipY=False, FBUrls=[], jsonRep=None):
if not skipFB:
# collect user input
console.prompt('Enter the persons name to find on FB: ')
name = input('')
while not name:
console.prompt('Enter the persons name to find on FB: ')
name = input('')
else:
console.task('Skipping FB Search')
name = "Unknown"
console.prompt(
'How many jitters, higher is better [max 100] (default=70): ')
num_jitters = input('')
if not num_jitters:
console.task('Settings jitters to 70')
num_jitters = 70
num_jitters = int(num_jitters)
if num_jitters > 100:
console.subfailure('Dude wtf?!')
num_jitters = 100
console.subfailure('Using 100 jitters...')
if not skipFB:
# grab profile urls
f = FBGrabber(name)
f.grabData()
# do face recognition on those profile images
r = FaceRecog(f.getProfileLinks(),
f.getProfileImages(),
num_jitters=num_jitters)
r.loadKnown(name)
profile_links, profile_imgs = r.getValidLinksAndImg(name)
console.section('Result')
console.task('Found the following Profiles:')
for i in range(len(profile_links)):
console.subtask(profile_links[i])
else:
if len(FBUrls) > 0:
f = FBProfileGrabber(FBUrls)
img_urls = f.grabLinks()
#FBURLS are our profile links synchron with img_urls
# so FBURLS[0] <=> img_urls[0]
r = FaceRecog(FBUrls, img_urls, num_jitters=num_jitters)
r.loadKnown(name)
profile_links, profile_imgs = r.getValidLinksAndImg(name)
console.section('Result')
console.task('Found the following Profiles:')
for i in range(len(profile_links)):
console.subtask(profile_links[i])
else:
profile_links = []
profile_imgs = []
# google reverse image search on profile pics
g = GoogleGrabber()
for img in profile_imgs:
g.collectLinks(img)
# google reverse image search on reference pic
g.collectLinksLocal()
rev_links, predictions = g.finish()
#TODO: Fix yandex search
#if not skipY:
if False:
yandex = YandexGrabber()
for img in profile_imgs:
yandex.collectLinks(img)
yandex.collectLinksLocal()
#add to rev_links
for e in yandex.finish():
rev_links.append(e)
else:
console.task('Skipping Yandex Search')
rev_links = list(set(rev_links))
instaNames = parseInstaUsername(filterInstaLinks(rev_links))
validatedInstaNames = []
console.section("Validating Instagram Profiles")
for un in instaNames:
console.task("Validating Profile: '{0}'".format(un))
if validateInstaUser(un, num_jitters):
validatedInstaNames.append(un)
raider_img_list = profile_imgs
for v in validatedInstaNames:
l = getInstaLinks(v)
for li in l:
raider_img_list.append(li)
if len(raider_img_list) <= 0:
console.failure('No Links found...')
else:
console.task('RIP Imageraider')
rev_links = list(set(rev_links))
predictions = list(set(predictions))
console.section('Links')
print(rev_links)
console.section('Predictions')
try:
predictions = [x.lower() for x in predictions]
except:
predictions = []
print(predictions)
presentResult(predictions)
for pl in profile_links:
rev_links.append(pl)
rev_links = list(set(rev_links))
#estimate age
ageEstimator = PictrievGrabber()
if len(validatedInstaNames) > 0:
for v in validatedInstaNames:
l = getInstaLinks(v)
if len(l) >= cfg.instaLimit():
l = l[:cfg.instaLimit()]
for li in l:
ageEstimator.collectAges(li)
age = ageEstimator.finish()
else:
console.failure('No Instagram Images to upload...')
#ageEstimator.finish()
age = "Unknown"
if jsonRep:
console.section("Dumping JSON Report")
makeJSONReport(name, rev_links, predictions, validatedInstaNames, age,
jsonRep)
else:
console.section("Creating PDF Report")
makeReport(name, rev_links, predictions, validatedInstaNames, age)
p = os.path.join(tempfile.gettempdir(), 'imageraider')
if os.path.isdir(p):
pathlist = Path(p).glob('**/*')
for path in pathlist:
s_p = str(path)
os.remove(s_p)
console.task("KTHXBYE")