Skip to content
GitLab
    • Explore Projects Groups Snippets
Projects Groups Snippets
  • /
  • Help
    • Help
    • Support
    • Community forum
    • Submit feedback
    • Contribute to GitLab
  • Sign in / Register
  • C csvkit
  • Project information
    • Project information
    • Activity
    • Labels
    • Members
  • Repository
    • Repository
    • Files
    • Commits
    • Branches
    • Tags
    • Contributors
    • Graph
    • Compare
  • Issues 61
    • Issues 61
    • List
    • Boards
    • Service Desk
    • Milestones
  • Merge requests 4
    • Merge requests 4
  • CI/CD
    • CI/CD
    • Pipelines
    • Jobs
    • Schedules
  • Deployments
    • Deployments
    • Environments
    • Releases
  • Packages and registries
    • Packages and registries
    • Package Registry
    • Infrastructure Registry
  • Monitor
    • Monitor
    • Incidents
  • Analytics
    • Analytics
    • Value stream
    • CI/CD
    • Repository
  • Wiki
    • Wiki
  • Snippets
    • Snippets
  • Activity
  • Graph
  • Create a new issue
  • Jobs
  • Commits
  • Issue Boards
Collapse sidebar
  • wireservice
  • csvkit
  • Merge requests
  • !865

Option to csvsql allowing custom SQL be executed before or after DB inserts

  • Review changes

  • Download
  • Email patches
  • Plain diff
Closed Administrator requested to merge github/fork/jest/master into master 8 years ago
  • Overview 4
  • Commits 1
  • Pipelines 0
  • Changes 1

Created by: jest

My use case: inserting into IDENTITY columns in SQL Server is not possible, unless you execute SET IDENTITY_INSERT <tablename> ON in the same transaction before. So, to allow for --insert in csvsql I have to execute this SQL before the inserts themself.

This PR adds --execute-before-insert and --execute-after-insert command line options.

Compare
  • master (base)

and
  • latest version
    9b5bfc9f
    1 commit, 2 years ago

1 file
+ 8
- 0

    Preferences

    File browser
    Compare changes
csvkit/utilities/csvsql.py
+ 8
- 0
  • View file @ 9b5bfc9f


@@ -44,6 +44,10 @@ class CSVSQL(CSVKitUtility):
help='Drop the table before creating.')
self.argparser.add_argument('--db-schema', dest='db_schema',
help='Optional name of database schema to create table(s) in.')
self.argparser.add_argument('--execute-before-insert', dest='execute_before_insert',
help='SQL to execute before inserts are performed')
self.argparser.add_argument('--execute-after-insert', dest='execute_after_insert',
help='SQL to execute after inserts are performed')
self.argparser.add_argument('-y', '--snifflimit', dest='sniff_limit', type=int,
help='Limit CSV dialect sniffing to the specified number of bytes. Specify "0" to disable sniffing entirely.')
self.argparser.add_argument('-I', '--no-inference', dest='no_inference', action='store_true',
@@ -135,6 +139,8 @@ class CSVSQL(CSVKitUtility):
if table:
if self.connection:
if self.args.execute_before_insert:
self.connection.execute(self.args.execute_before_insert)
table.to_sql(
self.connection,
table_name,
@@ -146,6 +152,8 @@ class CSVSQL(CSVKitUtility):
db_schema=self.args.db_schema,
constraints=not self.args.no_constraints
)
if self.args.execute_after_insert:
self.connection.execute(self.args.execute_after_insert)
# Output SQL statements
else:
0 Assignees
None
Assign to
0 Reviewers
None
Request review from
Labels
0
None
0
None
    Assign labels
  • Manage project labels

Milestone
No milestone
None
None
Time tracking
No estimate or time spent
Lock merge request
Unlocked
0
0 participants
Reference: donnemartin/system-design-primer!116
Source branch: github/fork/jest/master

Menu

Explore Projects Groups Snippets