Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- #!/usr/bin/env python3
- import sys
- query_file = sys.argv[1]
- csv_file = query_file + '.csv'
- def main():
- with open(csv_file, 'w') as csv:
- with open(query_file, 'r') as qf:
- for line in nonblank_lines(qf):
- if 'INSERT' in line:
- parens = 0
- for char in line:
- if parens >= 1 and not(char == ')'):
- csv.write(char)
- if char == '(':
- parens += 1
- if char == ')':
- parens -= 1
- if char == ')' and parens == 0:
- csv.write('\n')
- def nonblank_lines(qf):
- for l in qf:
- line = l.rstrip()
- if line:
- yield line
- if __name__ == "__main__":
- main()
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement