Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- # -*- coding: utf-8 -*-
- import os
- import sys
- def read_query(filename):
- with open(filename) as fp:
- return ''.join(fp.readlines())
- def create_folder(name):
- folder = './' + name
- if not os.path.isdir(folder):
- print 'Create folder %s' % folder
- os.mkdir(folder)
- def create_result(name, filename, qs):
- folder = './' + name
- create_folder(folder)
- for index, q in enumerate(qs):
- filename = 'query-%d.sql' % index
- with open(folder + '/' + filename, 'w') as fp:
- fp.write(q.strip() + '\n')
- def run(filename, name, cond_column, min_id, max_id, size):
- query = read_query(filename)
- qs = []
- while min_id < max_id:
- next_id = min_id + size
- cond_query = "AND %s >= %d AND %s < %d" % (
- cond_column, min_id, cond_column, next_id)
- qs.append(query + cond_query)
- min_id = next_id
- create_result(name, filename, qs)
- if __name__ == '__main__':
- name = sys.argv[1]
- filename = sys.argv[2]
- cond_columnd = sys.argv[3]
- min_id = int(sys.argv[4])
- max_id = int(sys.argv[5])
- size = int(sys.argv[6])
- run(filename, name, cond_columnd, min_id, max_id, size)
Add Comment
Please, Sign In to add comment