jpt-archive/mongoprof
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

flake8

Browse Source
This commit is contained in:
James Turk 2012-06-26 01:21:02 -04:00
parent 60aa0772e9
commit 9b48b8ea81
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
mongoprof.py
View File

@ -9,8 +9,10 @@ import argparse
from pymongo import Connection from pymongo import Connection
from termcolor import colored from termcolor import colored
# global quit monitor
quit = False quit = False
def watch(dbname, refresh): def watch(dbname, refresh):
global quit global quit
db = getattr(Connection('localhost'), dbname) db = getattr(Connection('localhost'), dbname)
@ -41,7 +43,8 @@ def watch(dbname, refresh):
elif e['op'] == 'update': elif e['op'] == 'update':
output.append(colored('update {query}'.format(**e), 'green')) output.append(colored('update {query}'.format(**e), 'green'))
elif e['op'] == 'getmore': elif e['op'] == 'getmore':
output.append(colored('getmore {0}'.format(e.get('query', '')), 'grey')) output.append(colored('getmore {0}'.format(e.get('query', '')),
'grey'))
elif e['op'] == 'command': elif e['op'] == 'command':
output.append(colored('{command}'.format(**e), 'cyan')) output.append(colored('{command}'.format(**e), 'cyan'))
else: else:
@ -64,6 +67,7 @@ def watch(dbname, refresh):
last_ts = e['ts'] last_ts = e['ts']
time.sleep(refresh) time.sleep(refresh)
def main(): def main():
parser = argparse.ArgumentParser(description='watch mongo queries') parser = argparse.ArgumentParser(description='watch mongo queries')
parser.add_argument('dbname', help='name of database to watch') parser.add_argument('dbname', help='name of database to watch')