itw-python-builder 0.2.4__tar.gz → 0.2.6__tar.gz

{itw_python_builder-0.2.4 → itw_python_builder-0.2.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: itw_python_builder
-Version: 0.2.4
+Version: 0.2.6
 Summary: Standardized Django deployment pipeline with Docker, testing, and SonarQube integration
 Author-email: IT-Works <contact@it-works.io>
 License: MIT
@@ -24,6 +24,7 @@ Requires-Dist: invoke>=2.0.0
 Requires-Dist: pylint>=3.0.0
 Requires-Dist: pylint-django>=2.5.0
 Requires-Dist: python-decouple>=3.8
+Requires-Dist: requests>=2.28.0
 Dynamic: license-file
 
 # ITW Python Builder

itw_python_builder-0.2.6/itw_python_builder/_pyruntime/sitecustomize.py

@@ -0,0 +1,27 @@
+def _install():
+    try:
+        from django.db.backends.postgresql import creation as _c
+    except Exception:
+        return
+    cls = _c.DatabaseCreation
+    if getattr(cls, '_itw_terminate_patched', False):
+        return
+    _orig = cls._destroy_test_db
+
+    def _patched(self, test_database_name, verbosity):
+        try:
+            with self._nodb_cursor() as cursor:
+                cursor.execute(
+                    "SELECT pg_terminate_backend(pid) FROM pg_stat_activity "
+                    "WHERE datname = %s AND pid <> pg_backend_pid()",
+                    [test_database_name],
+                )
+        except Exception:
+            pass
+        return _orig(self, test_database_name, verbosity)
+
+    cls._destroy_test_db = _patched
+    cls._itw_terminate_patched = True
+
+
+_install()

{itw_python_builder-0.2.4 → itw_python_builder-0.2.6}/itw_python_builder/notify.py

@@ -1,11 +1,9 @@
 import os
-import smtplib
-import ssl
 import sys
-from email.mime.multipart import MIMEMultipart
-from email.mime.text import MIMEText
 from pathlib import Path
 
+import requests
+
 TEMPLATE_PATH = Path(__file__).parent / 'templates' / 'new_version_email.html'
 
 
@@ -21,14 +19,12 @@ def _env_getter():
     return lambda key, default=None: os.environ[key] if default is None else os.environ.get(key, default)
 
 
-def _read_smtp_config() -> dict:
+def _read_mailgun_config() -> dict:
     get = _env_getter()
     return {
-        'host': get('EMAIL_HOST', 'smtp.gmail.com'),
-        'port': int(get('EMAIL_PORT', '587')),
-        'user': get('EMAIL_HOST_USER'),
-        'password': get('EMAIL_HOST_PASSWORD'),
-        'use_tls': str(get('EMAIL_USE_TLS', 'True')).lower() == 'true',
+        'from_email': get('FROM_EMAIL'),
+        'base_url': get('MAILGUN_BASE_URL'),
+        'api_key': get('MAILGUN_API_KEY'),
     }
 
 
@@ -56,24 +52,28 @@ def send_new_version_email(version: str, commit_message: str = '') -> int:
         print('[notify] EMAIL_RECIPIENTS is empty — skipping email notification.')
         return 0
 
-    cfg = _read_smtp_config()
+    cfg = _read_mailgun_config()
     html_body = _render_template(version, commit_message)
 
-    msg = MIMEMultipart('alternative')
-    msg['Subject'] = f'itw-python-builder — Version i ri ({version})'
-    msg['From'] = cfg['user']
-    msg['To'] = ', '.join(recipients)
-    msg.attach(MIMEText(html_body, 'html', 'utf-8'))
-
-    context = ssl.create_default_context()
-    with smtplib.SMTP(cfg['host'], cfg['port']) as smtp:
-        if cfg['use_tls']:
-            smtp.starttls(context=context)
-        smtp.login(cfg['user'], cfg['password'])
-        smtp.sendmail(cfg['user'], recipients, msg.as_string())
-
-    print(f'[notify] Sent release email for {version} to {len(recipients)} recipient(s).')
-    return len(recipients)
+    try:
+        data = {
+            'from': cfg['from_email'],
+            'to': recipients,
+            'subject': f'itw-python-builder — Version i ri ({version})',
+            'html': html_body,
+        }
+        response = requests.post(
+            cfg['base_url'],
+            auth=("api", cfg['api_key']),
+            data=data,
+        )
+        if not response.ok:
+            raise RuntimeError(f'Mailgun API error {response.status_code}: {response.text}')
+        print(f'[notify] Sent release email for {version} to {len(recipients)} recipient(s).')
+        return len(recipients)
+    except Exception as e:
+        print(f'FAILED TO SEND EMAIL TO {str(recipients)}: {str(e)}')
+        return 0
 
 
 def _main() -> int:

itw_python_builder-0.2.6/itw_python_builder/task_utils.py

@@ -0,0 +1,266 @@
+import re
+
+from itw_python_builder.utils import _gitlab_api_call
+
+
+DEFAULT_ACCEPTANCE_CRITERIA = (
+    '- [ ] U kryen testet manuale në ndërfaqe\n'
+    '- [ ] U kryen testet automatike\n'
+    '- [ ] U trajtuan sugjerimet e SonarQube\n'
+    '- [ ] U kontrollua cilësia e kodit'
+)
+
+
+_RE_AC_HEADER = re.compile(r'(?im)^##\s*acceptance\s+criteria\s*$')
+_RE_COMMENT_HEADER = re.compile(r'(?im)^##\s*comment\s*$')
+_RE_AC_BLOCK = re.compile(
+    r'(?ims)^##\s*acceptance\s+criteria\s*$(.*?)^##\s*acceptance\s+criteria\s*$'
+)
+_RE_COMMENT_BLOCK = re.compile(
+    r'(?ims)^##\s*comment\s*$(.*?)^##\s*comment\s*$'
+)
+
+
+_COLOR_GREEN = '\033[92m'
+_COLOR_RED = '\033[91m'
+_COLOR_RESET = '\033[0m'
+
+
+def log_green(msg: str) -> None:
+    print(f'{_COLOR_GREEN}{msg}{_COLOR_RESET}')
+
+
+def log_red(msg: str) -> None:
+    print(f'{_COLOR_RED}{msg}{_COLOR_RESET}')
+
+
+def log_info(msg: str) -> None:
+    print(msg)
+
+
+def _list_group_projects(api_host: str, group_path: str, token: str) -> list:
+    from urllib.parse import quote
+    encoded = quote(group_path, safe='')
+    return _gitlab_api_call(
+        api_host,
+        f'/groups/{encoded}/projects?per_page=100&search=_pm',
+        token,
+    )
+
+
+def resolve_pm_repo(api_host: str, project_path: str, token: str) -> list:
+    """Find _pm repos by walking the namespace upward from project_path."""
+    parts = project_path.split('/')
+    if len(parts) < 2:
+        raise RuntimeError(f'Project path {project_path!r} has no parent group.')
+    parts.pop()
+    while parts:
+        namespace = '/'.join(parts)
+        try:
+            projects = _list_group_projects(api_host, namespace, token)
+        except RuntimeError as exc:
+            if '404' in str(exc):
+                parts.pop()
+                continue
+            raise
+        matches = [p for p in projects if p.get('path', '').endswith('_pm')]
+        if matches:
+            return matches
+        parts.pop()
+    return []
+
+
+def create_gitlab_issue(api_host: str, project_id: int, payload: dict, token: str) -> dict:
+    return _gitlab_api_call(
+        api_host,
+        f'/projects/{project_id}/issues',
+        token,
+        method='POST',
+        payload=payload,
+    )
+
+
+def get_project_by_path(api_host: str, project_path: str, token: str) -> dict:
+    from urllib.parse import quote
+    encoded = quote(project_path, safe='')
+    return _gitlab_api_call(api_host, f'/projects/{encoded}', token)
+
+
+def find_project_milestone(api_host: str, project_id: int, title: str, token: str):
+    from urllib.parse import quote
+    encoded = quote(title, safe='')
+    results = _gitlab_api_call(
+        api_host,
+        f'/projects/{project_id}/milestones?search={encoded}',
+        token,
+    )
+    for m in results:
+        if m.get('title') == title:
+            return m
+    return None
+
+
+def create_project_milestone(
+    api_host: str,
+    project_id: int,
+    title: str,
+    start_date: str = None,
+    due_date: str = None,
+    token: str = None,
+) -> dict:
+    payload = {'title': title}
+    if start_date:
+        payload['start_date'] = start_date
+    if due_date:
+        payload['due_date'] = due_date
+    return _gitlab_api_call(
+        api_host,
+        f'/projects/{project_id}/milestones',
+        token,
+        method='POST',
+        payload=payload,
+    )
+
+
+def create_issue_note(api_host: str, project_id: int, issue_iid: int, body: str, token: str) -> dict:
+    return _gitlab_api_call(
+        api_host,
+        f'/projects/{project_id}/issues/{issue_iid}/notes',
+        token,
+        method='POST',
+        payload={'body': body},
+    )
+
+
+def _strip_milestone_wrap(raw: str) -> str:
+    raw = raw.strip()
+    if raw.startswith('%"') and raw.endswith('"'):
+        return raw[2:-1]
+    if raw.startswith('%'):
+        return raw[1:].strip().strip('"')
+    return raw
+
+
+def parse_md_issues(file_path: str) -> list:
+    """Parse a markdown file into structured issue dicts separated by '==='."""
+    with open(file_path, 'r', encoding='utf-8') as fp:
+        content = fp.read().replace('\r\n', '\n')
+    blocks = re.split(r'(?m)^\s*===\s*$', content)
+    issues = []
+    for idx, block in enumerate(blocks, start=1):
+        block = block.strip()
+        if not block:
+            continue
+        issues.append(_parse_single_issue(block, idx, file_path))
+    if not issues:
+        raise RuntimeError(f'No issues found in {file_path}.')
+    return issues
+
+
+def _parse_single_issue(block: str, idx: int, file_path: str) -> dict:
+    ac_headers = _RE_AC_HEADER.findall(block)
+    if len(ac_headers) == 1:
+        raise RuntimeError(
+            f'Block {idx} in {file_path}: found opening "## Acceptance Criteria" '
+            'but no closing "## Acceptance Criteria" tag.'
+        )
+    cm_headers = _RE_COMMENT_HEADER.findall(block)
+    if len(cm_headers) == 1:
+        raise RuntimeError(
+            f'Block {idx} in {file_path}: found opening "## Comment" '
+            'but no closing "## Comment" tag.'
+        )
+
+    lines = block.splitlines()
+    title = None
+    body_start = 0
+    for i, line in enumerate(lines):
+        if line.strip():
+            title = line.lstrip('#').strip()
+            body_start = i + 1
+            break
+    if not title:
+        raise RuntimeError(f'Block {idx} in {file_path}: missing title (first non-empty line).')
+
+    remaining = '\n'.join(lines[body_start:])
+
+    comment_text = None
+    if len(cm_headers) >= 2:
+        m = _RE_COMMENT_BLOCK.search(remaining)
+        if m:
+            comment_text = m.group(1).strip()
+            remaining = _RE_COMMENT_BLOCK.sub('', remaining, count=1)
+
+    acceptance_text = None
+    if len(ac_headers) >= 2:
+        m = _RE_AC_BLOCK.search(remaining)
+        if m:
+            acceptance_text = m.group(1).strip()
+            remaining = _RE_AC_BLOCK.sub('', remaining, count=1)
+
+    repo = None
+    milestone = None
+    milestone_start = None
+    milestone_end = None
+    assignee = None
+    has_estimate = False
+    has_due = False
+    body_lines = []
+
+    for line in remaining.splitlines():
+        stripped = line.strip()
+
+        m = re.match(r'^/repo\s+(\S+)\s*$', stripped, re.IGNORECASE)
+        if m:
+            repo = m.group(1)
+            continue
+        m = re.match(r'^/milestone-start\s+(\S+)\s*$', stripped, re.IGNORECASE)
+        if m:
+            milestone_start = m.group(1)
+            continue
+        m = re.match(r'^/milestone-end\s+(\S+)\s*$', stripped, re.IGNORECASE)
+        if m:
+            milestone_end = m.group(1)
+            continue
+        m = re.match(r'^/milestone\s+(.+)$', stripped, re.IGNORECASE)
+        if m:
+            milestone = _strip_milestone_wrap(m.group(1))
+            continue
+        m = re.match(r'^/assignee\s+(.+)$', stripped, re.IGNORECASE)
+        if m:
+            assignee = m.group(1).strip()
+            body_lines.append(f'/assign {assignee}')
+            continue
+        if re.match(r'^/estimate\s+\S+', stripped, re.IGNORECASE):
+            has_estimate = True
+        if re.match(r'^/due\s+\S+', stripped, re.IGNORECASE):
+            has_due = True
+
+        body_lines.append(line)
+
+    body = '\n'.join(body_lines).strip()
+    return {
+        'title': title,
+        'body': body,
+        'repo': repo,
+        'milestone': milestone,
+        'milestone_start': milestone_start,
+        'milestone_end': milestone_end,
+        'assignee': assignee,
+        'has_estimate': has_estimate,
+        'has_due': has_due,
+        'has_acceptance_criteria': acceptance_text is not None,
+        'acceptance_criteria': acceptance_text,
+        'has_comment': comment_text is not None,
+        'comment': comment_text,
+    }
+
+
+def build_issue_description(issue: dict) -> str:
+    """Compose the final GitLab issue description with acceptance criteria section."""
+    parts = []
+    if issue['body']:
+        parts.append(issue['body'])
+    ac = issue['acceptance_criteria'] if issue['has_acceptance_criteria'] else DEFAULT_ACCEPTANCE_CRITERIA
+    parts.append(f'## Acceptance Criteria\n{ac}')
+    return '\n\n'.join(parts)

{itw_python_builder-0.2.4 → itw_python_builder-0.2.6}/itw_python_builder/tasks.py

@@ -8,6 +8,7 @@ from datetime import datetime
 from itw_python_builder.ssr_tasks import *  # noqa: F401,F403  — exposes ssr-init
 from itw_python_builder.utils import (
     detect_and_activate_venv,
+    terminate_stale_test_db_sessions,
     load_env,
     get_current_branch,
     check_branch,
@@ -28,9 +29,19 @@ from itw_python_builder.utils import (
     TOKEN_CACHE_PATH,
     maybe_override_release_version,
     ensure_gitlab_token,
+)
+from itw_python_builder.task_utils import (
     resolve_pm_repo,
-    create_gitlab_task,
+    create_gitlab_issue,
     parse_md_issues,
+    build_issue_description,
+    get_project_by_path,
+    find_project_milestone,
+    create_project_milestone,
+    create_issue_note,
+    log_green,
+    log_red,
+    log_info,
 )
 from itw_python_builder.version import Version
 
@@ -113,7 +124,7 @@ def logout(ctx: Context):
 
 @task(name='task')
 def create_task(ctx: Context, file=None):
-    """Create GitLab tasks in the group's _pm repo from a markdown file (--file path.md)."""
+    """Create GitLab issues in the group's _pm repo from a markdown file (--file path.md)."""
     if not file:
         raise RuntimeError('Missing --file path to markdown file.')
     if not file.lower().endswith('.md'):
@@ -123,51 +134,126 @@ def create_task(ctx: Context, file=None):
     if not os.path.isdir(os.path.join(os.getcwd(), '.git')):
         raise RuntimeError('Not a git repository (no .git folder here).')
 
-    issues = parse_md_issues(file)
-    print(f'[itw] Parsed {len(issues)} task(s) from {file}.')
+    try:
+        issues = parse_md_issues(file)
+    except RuntimeError as exc:
+        log_red(str(exc))
+        raise
+    print(f'[itw] Parsed {len(issues)} issue(s) from {file}.')
 
     token = ensure_gitlab_token(ctx)
     host, project_path = _parse_gitlab_remote(ctx)
     api_host = _GITLAB_API_HOST_MAP.get(host, host)
 
-    print(f'[itw] Looking for _pm repos under {project_path!r}...')
+    for idx, issue in enumerate(issues, start=1):
+        print(f'\n[itw] Issue {idx}/{len(issues)}: {issue["title"]}')
+        try:
+            project = _resolve_target_project(api_host, project_path, issue, token)
+
+            if issue['has_acceptance_criteria']:
+                log_info('Using inputed acceptance criteria')
+            else:
+                log_green(
+                    'Default acceptance criteria applied. To define a custom acceptance criteria, '
+                    'define the section ## Acceptance Criteria in your md file and make sure to end '
+                    'it with ## Acceptance Criteria'
+                )
+
+            if issue['assignee']:
+                log_info(f'This issue will be assigned to {issue["assignee"]}')
+            else:
+                log_red('No asignee is defined for this issue.Use /assignee in you md file to define an asignee')
+
+            if not issue['has_estimate']:
+                log_red('There is no estimation for this issue. Use /estimate in you md file to define an estimation')
+
+            if not issue['has_due']:
+                log_red('No due date is defined for this issue. Use /due in you md file to define a due date')
+
+            milestone_id = _resolve_milestone(api_host, project['id'], issue, token)
+
+            description = build_issue_description(issue)
+            payload = {'title': issue['title'], 'description': description}
+            if milestone_id is not None:
+                payload['milestone_id'] = milestone_id
+
+            result = create_gitlab_issue(api_host, project['id'], payload, token)
+            print(f'[itw] Created: {result["web_url"]}')
+
+            if issue['has_comment']:
+                create_issue_note(api_host, project['id'], result['iid'], issue['comment'], token)
+                log_info('Comments added sucessfully')
+            else:
+                log_green(
+                    'No coments found on this issue. To leave a comment define the section '
+                    '## Comment section in your md file and make sure to end it with ## Comment'
+                )
+        except Exception as exc:
+            log_red(f'Failed at issue {idx}/{len(issues)} ({issue["title"]!r}): {exc}')
+            print(f'[itw] {idx - 1} issue(s) were created successfully before this failure.')
+            sys.exit(1)
+
+
+def _resolve_target_project(api_host: str, project_path: str, issue: dict, token: str) -> dict:
+    if issue['repo']:
+        log_info(f'Using {issue["repo"]} repo')
+        return get_project_by_path(api_host, issue['repo'], token)
+
+    log_green(f'Using current git repo {project_path}')
     candidates = resolve_pm_repo(api_host, project_path, token)
     if not candidates:
         raise RuntimeError(
             f'No _pm repo found anywhere in the namespace of {project_path!r}.'
         )
-
     if len(candidates) == 1:
         chosen = candidates[0]
         print(f'[itw] Using _pm repo: {chosen["path_with_namespace"]}')
-    else:
-        if not sys.stdin.isatty():
-            names = ', '.join(p['path_with_namespace'] for p in candidates)
-            raise RuntimeError(
-                f'{len(candidates)} _pm repos found ({names}) and stdin is not a TTY — cannot prompt.'
-            )
-        print(f'[itw] Found {len(candidates)} _pm repos:')
-        for i, p in enumerate(candidates, start=1):
-            print(f'  [{i}] {p["path_with_namespace"]}')
-        while True:
-            raw = input(f'Pick one (1-{len(candidates)}): ').strip()
-            try:
-                idx = int(raw)
-                if 1 <= idx <= len(candidates):
-                    chosen = candidates[idx - 1]
-                    break
-            except ValueError:
-                pass
-            print('Invalid choice, try again.')
-
-    for idx, (title, body) in enumerate(issues, start=1):
+        return chosen
+    if not sys.stdin.isatty():
+        names = ', '.join(p['path_with_namespace'] for p in candidates)
+        raise RuntimeError(
+            f'{len(candidates)} _pm repos found ({names}) and stdin is not a TTY — cannot prompt.'
+        )
+    print(f'[itw] Found {len(candidates)} _pm repos:')
+    for i, p in enumerate(candidates, start=1):
+        print(f'  [{i}] {p["path_with_namespace"]}')
+    while True:
+        raw = input(f'Pick one (1-{len(candidates)}): ').strip()
         try:
-            result = create_gitlab_task(api_host, chosen['id'], title, body, token)
-        except Exception as exc:
-            print(f'[itw] Failed at task {idx}/{len(issues)} ({title!r}): {exc}')
-            print(f'[itw] {idx - 1} task(s) were created successfully before this failure.')
-            raise
-        print(f'[itw] Created: {result["web_url"]}')
+            idx = int(raw)
+            if 1 <= idx <= len(candidates):
+                return candidates[idx - 1]
+        except ValueError:
+            pass
+        print('Invalid choice, try again.')
+
+
+def _resolve_milestone(api_host: str, project_id: int, issue: dict, token: str):
+    title = issue['milestone']
+    if not title:
+        log_red(
+            'There is no milestone defined for this issue. '
+            'Use /milestone in your md file to define a milestone'
+        )
+        return None
+
+    existing = find_project_milestone(api_host, project_id, title, token)
+    if existing:
+        log_info(f'Using existing milestone {title}')
+        return existing['id']
+
+    start = issue['milestone_start']
+    end = issue['milestone_end']
+    if start and end:
+        created = create_project_milestone(api_host, project_id, title, start, end, token)
+        log_green(f'Created new milestone {title} | {start}-{end}')
+    else:
+        created = create_project_milestone(api_host, project_id, title, token=token)
+        log_green(
+            f'Created new milestone {title} | No start date or end date is defined for this '
+            'new milestone. Use /milestone-start and /milestone-end in your md file to define them '
+        )
+    return created['id']
 
 
 def build_frontend(ctx: Context, branch: str, ssr: bool = False) -> None:
@@ -308,7 +394,7 @@ def tag_build_push(ctx: Context, version: Version, skip_pipeline: bool = False,
     install_sigint_guard()
     project_type = detect_project_type()
     if not skip_pipeline:
-        step('tests', test, ctx)
+        step('tests', test, ctx, warn=False)
         step('analyze', analyze, ctx)
     step('tag', tag, ctx, version)
     if project_type == 'backend':
@@ -442,14 +528,26 @@ def release(ctx: Context, skip_pipeline=False, ssr=False) -> None:
 
 
 @task
-def test(ctx: Context, settings='backend.test_settings', warn=False):
-    """Run tests with coverage. --warn lets the pipeline continue and print coverage on failure."""
+def test(ctx: Context, settings='backend.test_settings', warn=True):
+    """Run tests with coverage."""
     if detect_project_type() == 'frontend':
         ctx.run('npx ng test --no-watch --code-coverage --browsers=ChromeHeadlessNoSandbox', warn=warn)
         print("✓ Tests completed")
         return
     detect_and_activate_venv()
-    ctx.run(f'python -m coverage run manage.py test --settings={settings} --noinput -v 2', warn=warn)
+    load_env(ctx)
+    terminate_stale_test_db_sessions(ctx, settings)
+    patch_dir = str(Path(__file__).parent / '_pyruntime')
+    existing_pp = os.environ.get('PYTHONPATH', '')
+    injected_pp = f'{patch_dir}:{existing_pp}' if existing_pp else patch_dir
+    try:
+        ctx.run(
+            f'python -m coverage run manage.py test --settings={settings} --noinput -v 2',
+            warn=warn,
+            env={'PYTHONPATH': injected_pp},
+        )
+    finally:
+        terminate_stale_test_db_sessions(ctx, settings)
     ctx.run('python -m coverage report -m')
     ctx.run('python -m coverage xml -o coverage.xml')
     print("✓ Tests completed")
@@ -510,6 +608,10 @@ def analyze(ctx: Context):
 
     if not sonar_host or not sonar_token:
         raise RuntimeError('SONAR_HOST_URL and SONAR_TOKEN environment variables must be set.')
+
+    sonar_cache_dir = os.path.join(os.path.expanduser('~'), '.sonar', 'cache')
+    os.makedirs(sonar_cache_dir, exist_ok=True)
+
     ctx.run(
         f'podman run --rm -e SONAR_HOST_URL={sonar_host} -e SONAR_TOKEN={sonar_token} -e GIT_DEPTH={git_depth} --user=root -v "$(pwd)":/usr/src:Z -v $HOME/.sonar/cache:/opt/sonar-scanner/.sonar/cache:Z docker.io/sonarsource/sonar-scanner-cli:latest -Dsonar.projectVersion={sonar_version}')
 
@@ -543,7 +645,7 @@ def pipelinelocal(ctx: Context, settings='backend.test_settings', pylintrc=None)
     print("\n" + "=" * 60)
     print(f"Step {'2' if project_type == 'backend' else '1'}: Running tests...")
     print("=" * 60)
-    step('tests', test, ctx, settings, warn=True)
+    step('tests', test, ctx, settings, warn=False)
 
     print("\n" + "=" * 60)
     print(f"Step {'3' if project_type == 'backend' else '2'}: Running SonarQube analysis...")

{itw_python_builder-0.2.4 → itw_python_builder-0.2.6}/itw_python_builder/utils.py

@@ -2,8 +2,10 @@ import io
 import json
 import os
 import re
+import shlex
 import signal
 import sys
+import textwrap
 from pathlib import Path
 
 from invoke import Context
@@ -120,6 +122,60 @@ def detect_and_activate_venv():
     _venv_activated = True
 
 
+def terminate_stale_test_db_sessions(ctx: Context, settings_module: str) -> None:
+    script = textwrap.dedent(f'''
+        import os, sys
+        os.environ["DJANGO_SETTINGS_MODULE"] = {settings_module!r}
+        try:
+            import django
+            django.setup()
+            from django.conf import settings as _s
+        except Exception as _e:
+            print(f"[itw] session-cleanup skipped: cannot load Django settings ({{_e}})")
+            sys.exit(0)
+        try:
+            import psycopg2 as _pg
+        except ImportError:
+            try:
+                import psycopg as _pg
+            except ImportError:
+                print("[itw] session-cleanup skipped: no psycopg installed")
+                sys.exit(0)
+        for _alias, _cfg in _s.DATABASES.items():
+            if "postgresql" not in _cfg.get("ENGINE", ""):
+                continue
+            _test_name = (_cfg.get("TEST") or {{}}).get("NAME") or f"test_{{_cfg.get('NAME', '')}}"
+            if not _test_name or _test_name == "test_":
+                continue
+            try:
+                _conn = _pg.connect(
+                    host=_cfg.get("HOST") or "localhost",
+                    port=int(_cfg.get("PORT") or 5432),
+                    user=_cfg.get("USER") or None,
+                    password=_cfg.get("PASSWORD") or None,
+                    dbname="postgres",
+                    connect_timeout=5,
+                )
+                _conn.autocommit = True
+                _cur = _conn.cursor()
+                _cur.execute(
+                    "SELECT pg_terminate_backend(pid) FROM pg_stat_activity "
+                    "WHERE datname = %s AND pid <> pg_backend_pid()",
+                    [_test_name],
+                )
+                _rows = _cur.fetchall()
+                _cur.close()
+                _conn.close()
+                _killed = len(_rows)
+                if _killed > 0:
+                    print(f"[itw] Terminated {{_killed}} stale session(s) on {{_test_name}}")
+            except Exception as _e:
+                print(f"[itw] session-cleanup skipped for {{_test_name}}: {{_e}}")
+                continue
+    ''').strip()
+    ctx.run(f'python -c {shlex.quote(script)}', warn=True)
+
+
 def _parse_angular_env_file(path: str) -> dict:
     """Extract string-valued keys from an Angular environment.ts object literal."""
     with open(path, 'r', encoding='utf-8') as fp:
@@ -494,72 +550,3 @@ def ensure_gitlab_token(ctx: Context) -> str:
     return token
 
 
-def _list_group_projects(api_host: str, group_path: str, token: str) -> list:
-    """List up to 100 projects in a group matching '_pm'."""
-    from urllib.parse import quote
-    encoded = quote(group_path, safe='')
-    return _gitlab_api_call(
-        api_host,
-        f'/groups/{encoded}/projects?per_page=100&search=_pm',
-        token,
-    )
-
-
-def resolve_pm_repo(api_host: str, project_path: str, token: str) -> list:
-    """Find _pm repos by walking the namespace upward from project_path."""
-    parts = project_path.split('/')
-    if len(parts) < 2:
-        raise RuntimeError(f'Project path {project_path!r} has no parent group.')
-    parts.pop()
-    while parts:
-        namespace = '/'.join(parts)
-        try:
-            projects = _list_group_projects(api_host, namespace, token)
-        except RuntimeError as exc:
-            if '404' in str(exc):
-                parts.pop()
-                continue
-            raise
-        matches = [p for p in projects if p.get('path', '').endswith('_pm')]
-        if matches:
-            return matches
-        parts.pop()
-    return []
-
-
-def create_gitlab_task(api_host: str, project_id: int, title: str, description: str, token: str) -> dict:
-    """Create a GitLab task in the given project."""
-    return _gitlab_api_call(
-        api_host,
-        f'/projects/{project_id}/issues',
-        token,
-        method='POST',
-        payload={'title': title, 'description': description, 'issue_type': 'task'},
-    )
-
-
-def parse_md_issues(file_path: str) -> list:
-    """Parse a markdown file into (title, body) blocks separated by '==='."""
-    with open(file_path, 'r', encoding='utf-8') as fp:
-        content = fp.read().replace('\r\n', '\n')
-    blocks = re.split(r'(?m)^\s*===\s*$', content)
-    issues = []
-    for idx, block in enumerate(blocks, start=1):
-        block = block.strip()
-        if not block:
-            continue
-        lines = block.splitlines()
-        title = None
-        body_start = 0
-        for i, line in enumerate(lines):
-            if line.strip():
-                title = line.lstrip('#').strip()
-                body_start = i + 1
-                break
-        if not title:
-            raise RuntimeError(f'Block {idx} in {file_path}: missing title (first non-empty line).')
-        body = '\n'.join(lines[body_start:]).strip()
-        issues.append((title, body))
-    if not issues:
-        raise RuntimeError(f'No issues found in {file_path}.')
-    return issues

{itw_python_builder-0.2.4 → itw_python_builder-0.2.6}/itw_python_builder.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: itw_python_builder
-Version: 0.2.4
+Version: 0.2.6
 Summary: Standardized Django deployment pipeline with Docker, testing, and SonarQube integration
 Author-email: IT-Works <contact@it-works.io>
 License: MIT
@@ -24,6 +24,7 @@ Requires-Dist: invoke>=2.0.0
 Requires-Dist: pylint>=3.0.0
 Requires-Dist: pylint-django>=2.5.0
 Requires-Dist: python-decouple>=3.8
+Requires-Dist: requests>=2.28.0
 Dynamic: license-file
 
 # ITW Python Builder

{itw_python_builder-0.2.4 → itw_python_builder-0.2.6}/itw_python_builder.egg-info/SOURCES.txt

@@ -6,6 +6,7 @@ itw_python_builder/__init__.py
 itw_python_builder/cli.py
 itw_python_builder/notify.py
 itw_python_builder/ssr_tasks.py
+itw_python_builder/task_utils.py
 itw_python_builder/tasks.py
 itw_python_builder/utils.py
 itw_python_builder/version.py
@@ -15,6 +16,7 @@ itw_python_builder.egg-info/dependency_links.txt
 itw_python_builder.egg-info/entry_points.txt
 itw_python_builder.egg-info/requires.txt
 itw_python_builder.egg-info/top_level.txt
+itw_python_builder/_pyruntime/sitecustomize.py
 itw_python_builder/templates/new_version_email.html
 itw_python_builder/templates/server.sitemap.snippet.ts
 itw_python_builder/templates/sitemap.routes.ts

{itw_python_builder-0.2.4 → itw_python_builder-0.2.6}/itw_python_builder.egg-info/requires.txt

@@ -2,3 +2,4 @@ invoke>=2.0.0
 pylint>=3.0.0
 pylint-django>=2.5.0
 python-decouple>=3.8
+requests>=2.28.0

{itw_python_builder-0.2.4 → itw_python_builder-0.2.6}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "itw_python_builder"
-version = "0.2.4"
+version = "0.2.6"
 description = "Standardized Django deployment pipeline with Docker, testing, and SonarQube integration"
 readme = "README.md"
 requires-python = ">=3.10"
@@ -29,6 +29,7 @@ dependencies = [
     "pylint>=3.0.0",
     "pylint-django>=2.5.0",
     "python-decouple>=3.8",
+    "requests>=2.28.0",
 ]
 
 [project.urls]
@@ -40,7 +41,7 @@ Issues = "https://git.it-works.io/"
 include-package-data = true
 
 [tool.setuptools.package-data]
-itw_python_builder = [".pylintrc", "templates/*.ts", "templates/*.html"]
+itw_python_builder = [".pylintrc", "templates/*.ts", "templates/*.html", "_pyruntime/*.py"]
 
 [project.scripts]
 itw = "itw_python_builder.cli:main"