Guest User

Untitled

a guest
Mar 18th, 2018
93
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 1.17 KB | None | 0 0
  1. # -*- coding: utf-8 -*-
  2.  
  3. import os
  4. import sys
  5.  
  6. def read_query(filename):
  7. with open(filename) as fp:
  8. return ''.join(fp.readlines())
  9.  
  10.  
  11. def create_folder(name):
  12. folder = './' + name
  13. if not os.path.isdir(folder):
  14. print 'Create folder %s' % folder
  15. os.mkdir(folder)
  16.  
  17.  
  18. def create_result(name, filename, qs):
  19. folder = './' + name
  20. create_folder(folder)
  21.  
  22. for index, q in enumerate(qs):
  23. filename = 'query-%d.sql' % index
  24. with open(folder + '/' + filename, 'w') as fp:
  25. fp.write(q.strip() + '\n')
  26.  
  27.  
  28. def run(filename, name, cond_column, min_id, max_id, size):
  29. query = read_query(filename)
  30. qs = []
  31.  
  32. while min_id < max_id:
  33. next_id = min_id + size
  34.  
  35. cond_query = "AND %s >= %d AND %s < %d" % (
  36. cond_column, min_id, cond_column, next_id)
  37. qs.append(query + cond_query)
  38.  
  39. min_id = next_id
  40.  
  41. create_result(name, filename, qs)
  42.  
  43.  
  44. if __name__ == '__main__':
  45. name = sys.argv[1]
  46. filename = sys.argv[2]
  47. cond_columnd = sys.argv[3]
  48. min_id = int(sys.argv[4])
  49. max_id = int(sys.argv[5])
  50. size = int(sys.argv[6])
  51.  
  52. run(filename, name, cond_columnd, min_id, max_id, size)
Add Comment
Please, Sign In to add comment