amd/brain
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

fixing buggy code when looking for secrets, retrying same URL is ineffective

Browse source
This commit is contained in:
Andrew Davidson 2012-05-23 11:36:59 -07:00
parent daff5c46ca
commit f235cc8af4
1 changed files with 10 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
brain.py
View file

@ -37,17 +37,21 @@ conf['ext_render'] = ['.md','.txt','.jpg','.gif','.png']
conf['ext_edit'] = ['.md','.txt','.rb','.py','.pl','.sh']
def get_secrets(url = False):
if url:
query = 'SELECT base_url,username,password,id FROM `secrets` WHERE base_url LIKE "%' + url + '";'
else:
query = 'SELECT base_url,username,password,id FROM `secrets`;'
def generate_query():
if url:
query = 'SELECT base_url,username,password,id FROM `secrets` WHERE base_url LIKE "%' + url + '";'
else:
query = 'SELECT base_url,username,password,id FROM `secrets`;'
return query
cursor.execute(query)
cursor.execute(generate_query())
secrets = cursor.fetchall()
if not len(secrets) > 0:
print 'No secrets for ' + url + ' expanding search.'
url = get_domain(url)
cursor.execute(query)
print 'Trying ' + url + '.'
cursor.execute(generate_query())
secrets = cursor.fetchall()
return secrets