mirrors
/
pgcli
mirror of https://github.com/dbcli/pgcli
1
0
Fork 0
Code

black'd the code.

This commit is contained in:
ERYoung11 2022-08-31 21:48:06 -05:00
parent f5669e756b
commit 2850c83310
2 changed files with 14 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
pgcli/pgexecute.py
View File

@ -308,22 +308,22 @@ class PGExecute:
# sql parse doesn't split on a comment first + special
# so we're going to do it
sqltemp=[]
sqlarr=[]
sqltemp = []
sqlarr = []
if statement.startswith("--"):
sqltemp=statement.split("\n")
sqltemp = statement.split("\n")
sqlarr.append(sqltemp[0])
for i in sqlparse.split(sqltemp[1]):
sqlarr.append(i)
elif statement.startswith("/*"):
sqltemp=statement.split("*/")
sqltemp[0]=sqltemp[0]+"*/"
sqltemp = statement.split("*/")
sqltemp[0] = sqltemp[0] + "*/"
for i in sqlparse.split(sqltemp[1]):
sqlarr.append(i)
else:
sqlarr = sqlparse.split(statement)
sqlarr = sqlparse.split(statement)
# run each sql query
for sql in sqlarr:

10
tests/test_pgexecute.py
View File

@ -281,8 +281,11 @@ def test_execute_from_file_io_error(os, executor, pgspecial):
assert success == False
assert is_special == True
@dbtest
def test_execute_from_commented_file_that_executes_another_file(executor, pgspecial, tmpdir):
def test_execute_from_commented_file_that_executes_another_file(
executor, pgspecial, tmpdir
):
# https://github.com/dbcli/pgcli/issues/1336
sqlfile1 = tmpdir.join("test01.sql")
sqlfile1.write("-- asdf \n\\h")
@ -296,7 +299,8 @@ def test_execute_from_commented_file_that_executes_another_file(executor, pgspec
statement = "--comment\n\\h"
result = run(executor, statement, pgspecial=cli.pgspecial)
assert result != None
assert result[0].find("ALTER TABLE")
assert result[0].find("ALTER TABLE")
@dbtest
def test_execute_commented_first_line_and_special(executor, pgspecial, tmpdir):
@ -316,7 +320,7 @@ def test_execute_commented_first_line_and_special(executor, pgspecial, tmpdir):
statement = "/*comment\ncomment line2*/\nselect now();"
result = run(executor, statement, pgspecial=pgspecial)
assert result != None
assert result[1].find("now") >= 0
assert result[1].find("now") >= 0
statement = "--comment\n\\h"
result = run(executor, statement, pgspecial=pgspecial)