<s> """<STR_LIT>""" <EOL> from . . config import Config <EOL> from . . database import SqlStatement <EOL> from . . database import Table <EOL> from . . utils import constants as const <EOL> from . . utils . helpers import parse_path <EOL> from . qa_transform import QATransformStep <EOL> config = Config ( ) <EOL> class PrimaryKeyCheckStep ( QATransformStep ) : <EOL> """<STR_LIT>""" <EOL> def __init__ ( self , id , table_definition , script_arguments = None , <EOL> log_to_s3 = False , command = None , script = None , ** kwargs ) : <EOL> """<STR_LIT>""" <EOL> with open ( parse_path ( table_definition ) ) as f : <EOL> table_def_string = f . read ( ) <EOL> if script_arguments is None : <EOL> script_arguments = list ( ) <EOL> script_arguments . append ( <EOL> '<STR_LIT>' % Table ( SqlStatement ( table_def_string ) ) . sql ( ) ) <EOL> if log_to_s3 : <EOL> script_arguments . append ( '<STR_LIT>' ) <EOL> if script is None and command is None : <EOL> command = const . PK_CHECK_COMMAND <EOL> super ( PrimaryKeyCheckStep , self ) . __init__ ( <EOL> id = id , command = command , script = script , <EOL> script_arguments = script_arguments , ** kwargs ) </s>
