#! /usr/bin/env python
-import sys, os, tempfile, psycopg2
+import sys
+import os
+import tempfile
+import psycopg2
if len(sys.argv) != 3:
print('usage: mkauth DSTFN CONNSTR')
lines = []
for user, psw in curs.fetchall():
user = user.replace('"', '""')
- if not psw: psw = ''
+ if not psw:
+ psw = ''
psw = psw.replace('"', '""')
lines.append('"%s" "%s" ""\n' % (user, psw))
db.commit()
# if changed, replace data securely
if old != cur:
- fd, tmpfn = tempfile.mkstemp(dir = os.path.split(fn)[0])
+ fd, tmpfn = tempfile.mkstemp(dir=os.path.split(fn)[0])
f = os.fdopen(fd, 'w')
f.write(cur)
f.close()
#! /usr/bin/env python
-import sys, os, re, time, psycopg2
-import threading, thread, random
+import sys
+import os
+import re
+import time
+import psycopg2
+import threading
+import thread
+import random
n_thread = 100
-longtx = 0
-tx_sleep = 0
+longtx = False
tx_sleep = 8
conn_data = {
'connect_timeout': '5',
}
+
def get_connstr():
tmp = []
for k, v in conn_data.items():
tmp.append(k+'='+v)
return " ".join(tmp)
+
class WorkThread(threading.Thread):
def __init__(self):
threading.Thread.__init__(self)
def run(self):
try:
time.sleep(random.random() * 10.0)
- except: pass
+ except Exception:
+ pass
while 1:
try:
self.main_loop()
print d
try:
time.sleep(5)
- except: pass
+ except Exception:
+ pass
def main_loop(self):
db = psycopg2.connect(get_connstr())
if longtx:
db.commit()
+
def main():
print "connstr", get_connstr()
avg = cnt / dur
print "avg", avg
+
if __name__ == '__main__':
try:
main()