mirror of
https://github.com/boostorg/mysql.git
synced 2026-02-14 12:52:17 +00:00
34 lines
951 B
Python
34 lines
951 B
Python
#
|
|
# Copyright (c) 2019-2020 Ruben Perez Hidalgo (rubenperez038 at gmail dot com)
|
|
#
|
|
# Distributed under the Boost Software License, Version 1.0. (See accompanying
|
|
# file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
|
|
#
|
|
|
|
from subprocess import run
|
|
import os
|
|
import argparse
|
|
|
|
def get_executable_name(name):
|
|
if os.name == 'nt':
|
|
name += '.exe'
|
|
return name
|
|
|
|
def run_sql_file(fname):
|
|
print('Running SQL file: {}'.format(fname))
|
|
with open(fname, 'rb') as f:
|
|
sql = f.read()
|
|
run([get_executable_name('mysql'), '-u', 'root'], input=sql, check=True)
|
|
|
|
def main():
|
|
parser = argparse.ArgumentParser(description='Make MySQL run a .sql file')
|
|
parser.add_argument('file_name')
|
|
parser.add_argument('-s', '--skip-var', dest='skip')
|
|
args = parser.parse_args()
|
|
|
|
if args.skip is None or args.skip not in os.environ:
|
|
run_sql_file(args.file_name)
|
|
|
|
if __name__ == '__main__':
|
|
main()
|