diff --git a/HISTORY.rst b/HISTORY.rst index 31ce6cd..0be7772 100644 --- a/HISTORY.rst +++ b/HISTORY.rst @@ -17,6 +17,10 @@ History - |backward-incompatible| Drop support of Django 3.0 and prior versions. Only Django 3.1 and later are supported. +- Add `CID_SQL_STATEMENT_TEMPLATE` setting to customize the position + of the correlation relative to the original SQL statement. + Contributed by CauĂȘ Garcia Polimanti (@CaueP). + 2.3 (2022-06-13) ++++++++++++++++ diff --git a/docs/installation.rst b/docs/installation.rst index fd31266..7811940 100644 --- a/docs/installation.rst +++ b/docs/installation.rst @@ -244,6 +244,15 @@ a string with a ``cid`` format parameter: CID_SQL_COMMENT_TEMPLATE = 'correlation={cid}' +Also, you may change the position of the correlation id injected in +the statement by defining a ``CID_SQL_STATEMENT_TEMPLATE`` that is +a string with a ``cid`` and a ``sql`` format parameter: + +.. code-block:: python + + CID_SQL_STATEMENT_TEMPLATE = '/* {cid} */\n{sql}' + + Inclusion of the correlation id in templates -------------------------------------------- diff --git a/src/cid/cursor.py b/src/cid/cursor.py index 2531fd8..f7282c1 100644 --- a/src/cid/cursor.py +++ b/src/cid/cursor.py @@ -4,6 +4,7 @@ DEFAULT_CID_SQL_COMMENT_TEMPLATE = 'cid: {cid}' +DEFAULT_SQL_STATEMENT_TEMPLATE = '/* {cid} */\n{sql}' class CidCursorWrapper: @@ -31,12 +32,16 @@ def add_comment(self, sql): cid_sql_template = getattr( settings, 'CID_SQL_COMMENT_TEMPLATE', DEFAULT_CID_SQL_COMMENT_TEMPLATE ) + sql_statement_template = getattr( + settings, 'CID_SQL_STATEMENT_TEMPLATE', DEFAULT_SQL_STATEMENT_TEMPLATE + ) cid = get_cid() if not cid: return sql cid = cid.replace('/*', r'\/\*').replace('*/', r'\*\/') cid = cid_sql_template.format(cid=cid) - return f"/* {cid} */\n{sql}" + statement = sql_statement_template.format(cid=cid, sql=sql) + return statement # The following methods cannot be implemented in __getattr__, because the # code must run when the method is invoked, not just when it is accessed. diff --git a/tests/test_cursor.py b/tests/test_cursor.py index 5d187e2..f1a912f 100644 --- a/tests/test_cursor.py +++ b/tests/test_cursor.py @@ -33,6 +33,16 @@ def test_adds_comment_setting_overriden(self, get_cid): self.cursor_wrapper.add_comment("SELECT 1;") ) + @override_settings(CID_SQL_STATEMENT_TEMPLATE='{sql}\n/* {cid} */') + @mock.patch('cid.cursor.get_cid') + def test_adds_comment_with_statement_template_setting_overriden(self, get_cid): + get_cid.return_value = 'testing-cursor-after-sql-statement' + expected = "SELECT 1;\n/* cid: testing-cursor-after-sql-statement */" + self.assertEqual( + expected, + self.cursor_wrapper.add_comment("SELECT 1;") + ) + @mock.patch('cid.cursor.get_cid') def test_no_comment_when_cid_is_none(self, get_cid): get_cid.return_value = None