djresttoolkit 0.6.2__py3-none-any.whl → 0.8.0__py3-none-any.whl

README.md

@@ -85,7 +85,132 @@ python manage.py dbseed --count 10
 python manage.py dbseed --model User --seed 42
 ```
 
-### 2. EmailSender
+Here’s a **concise API reference** for your database flush management command for `djresttoolkit`:
+
+---
+
+### 2. DB Flush Command
+
+```python
+from djresttoolkit.management.commands import flush
+```
+
+#### `manage.py dbflush`
+
+Command to **delete all records** from the database for all models or a specific model and **reset auto-increment IDs**.
+
+#### Usage
+
+```bash
+python manage.py flush [--model ModelName] [--yes]
+```
+
+#### dbflush command options
+
+- `--model`: Name of the model to flush (case-sensitive, e.g., `User`). If omitted, flushes all models.
+- `--yes`: Skip confirmation prompt. Without this, the command asks for confirmation before deleting.
+
+#### dbflush command behavior
+
+- Deletes all records for the specified model or all models.
+- Resets primary key sequences for supported databases:
+
+  - PostgreSQL: `ALTER SEQUENCE ... RESTART WITH 1`
+  - SQLite: Deletes from `sqlite_sequence` table
+  - Others: Logs a warning (not implemented).
+- Uses transactions to ensure safe operations.
+
+#### dbflush command example
+
+```bash
+# Flush all models with confirmation
+python manage.py dbflush
+
+# Flush a specific model (User) with confirmation
+python manage.py dbflush --model User
+
+# Flush all models without prompt
+python manage.py dbflush --yes
+```
+
+#### Output
+
+```bash
+Flushed 10 records from model "User" and reset IDs.
+```
+
+or
+
+```bash
+Flushed 120 records from all models and reset IDs.
+```
+
+### 3. EnvBaseSettings
+
+```python
+from djresttoolkit.envconfig import EnvBaseSettings
+```
+
+#### `EnvBaseSettings`
+
+A **base settings class** for managing application configuration using:
+
+- YAML files (default `.environ.yaml`)
+- Environment variables (default `.env`)
+
+Supports **nested configuration** using double underscores (`__`) in environment variable names.
+
+#### Class Attributes
+
+| Attribute      | Type                 | Default         | Description                                                        |
+| -------------- | -------------------- | --------------- | ------------------------------------------------------------------ |
+| `env_file`     | `str`                | `.env`          | Environment variable file path.                                    |
+| `yaml_file`    | `str`                | `.environ.yaml` | YAML configuration file path.                                      |
+| `model_config` | `SettingsConfigDict` | —               | Pydantic settings configuration (file encoding, nested delimiter). |
+
+#### Methods
+
+#### `load(cls, *, env_file: str | None = None, ymal_file: str | None = None, warning: bool = True) -> EnvBaseSettings`
+
+Loads configuration from **YAML first**, then overrides with **environment variables**.
+
+#### Parameters
+
+- `env_file` — Optional custom `.env` file path.
+- `ymal_file` — Optional custom YAML file path.
+- `warning` — Emit a warning if YAML file is missing (default `True`).
+
+#### Returns
+
+- Instance of `EnvBaseSettings` (or subclass) with loaded configuration.
+
+#### Raises
+
+- `UserWarning` if YAML file not found and `warning=True`.
+
+### Usage Example
+
+```python
+from djresttoolkit.envconfig import EnvBaseSettings
+
+class EnvSettings(EnvBaseSettings):
+    debug: bool = False
+    database_url: str
+
+# Load settings
+settings = EnvSettings.load(warning=False)
+
+print(settings.debug)
+print(settings.database_url)
+```
+
+#### Features
+
+- Prioritizes `.env` variables over YAML.
+- Supports nested keys: `DATABASE__HOST` → `settings.database.host`.
+- Designed to be subclassed for project-specific settings.
+
+### 4. EmailSender
 
 ```python
 from djresttoolkit.mail import EmailSender, EmailContent, EmailTemplate
@@ -101,7 +226,7 @@ Send templated emails.
 EmailSender(email_content: EmailContent | EmailContentDict)
 ```
 
-#### Methods
+#### EmailSender Methods
 
 ```python
 send(to: list[str], exceptions: bool = False) -> bool
@@ -134,7 +259,7 @@ EmailSender(content).send(to=["user@example.com"])
 
 - `text`, `html` — template file paths
 
-### 3. Custom DRF Exception Handler
+### 5. Custom DRF Exception Handler
 
 ```python
 from djresttoolkit.views import exception_handler
@@ -148,12 +273,12 @@ A DRF exception handler that:
 - Adds throttling support (defaults to `AnonRateThrottle`).
 - Returns **429 Too Many Requests** with `retry_after` if throttle limit is exceeded.
 
-#### Parameters
+#### Exception Handler Parameters
 
 - `exc`: Exception object.
 - `context`: DRF context dictionary containing `"request"` and `"view"`.
 
-#### Returns
+#### Returns Type of Exception Handler
 
 - `Response` — DRF Response object (with throttling info if applicable), or `None`.
 
@@ -174,7 +299,7 @@ REST_FRAMEWORK = {
 - Tracks requests in cache and calculates `retry_after`.
 - Cleans expired timestamps automatically.
 
-### 4. Response Time Middleware
+### 6. Response Time Middleware
 
 ```python
 from djresttoolkit.middlewares import ResponseTimeMiddleware
@@ -192,7 +317,7 @@ ResponseTimeMiddleware(get_response: Callable[[HttpRequest], HttpResponse])
 
 - `get_response`: The next middleware or view callable.
 
-#### Usage
+#### Response Time Middleware Usage
 
 Add it to your Django `MIDDLEWARE` in `settings.py`:
 
@@ -221,7 +346,7 @@ X-Response-Time: 0.01234 seconds
 INFO: Request processed in 0.01234 seconds
 ```
 
-### 5. Throttle Utilities
+### 7. Throttle Utilities
 
 #### `ThrottleInfoJSONRenderer`
 

{djresttoolkit-0.6.2.dist-info → djresttoolkit-0.8.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: djresttoolkit
-Version: 0.6.2
+Version: 0.8.0
 Summary: A collection of Django and DRF utilities to simplify API development.
 Project-URL: Homepage, https://github.com/shaileshpandit141/djresttoolkit
 Project-URL: Documentation, https://shaileshpandit141.github.io/djresttoolkit
@@ -47,7 +47,9 @@ Classifier: Topic :: Utilities
 Classifier: Typing :: Typed
 Requires-Python: >=3.13
 Requires-Dist: faker>=37.5.3
+Requires-Dist: pydantic-settings>=2.10.1
 Requires-Dist: pydantic>=2.11.7
+Requires-Dist: pyyaml>=6.0.2
 Provides-Extra: dev
 Requires-Dist: mypy; extra == 'dev'
 Requires-Dist: pytest; extra == 'dev'
@@ -141,7 +143,132 @@ python manage.py dbseed --count 10
 python manage.py dbseed --model User --seed 42
 ```
 
-### 2. EmailSender
+Here’s a **concise API reference** for your database flush management command for `djresttoolkit`:
+
+---
+
+### 2. DB Flush Command
+
+```python
+from djresttoolkit.management.commands import flush
+```
+
+#### `manage.py dbflush`
+
+Command to **delete all records** from the database for all models or a specific model and **reset auto-increment IDs**.
+
+#### Usage
+
+```bash
+python manage.py flush [--model ModelName] [--yes]
+```
+
+#### dbflush command options
+
+- `--model`: Name of the model to flush (case-sensitive, e.g., `User`). If omitted, flushes all models.
+- `--yes`: Skip confirmation prompt. Without this, the command asks for confirmation before deleting.
+
+#### dbflush command behavior
+
+- Deletes all records for the specified model or all models.
+- Resets primary key sequences for supported databases:
+
+  - PostgreSQL: `ALTER SEQUENCE ... RESTART WITH 1`
+  - SQLite: Deletes from `sqlite_sequence` table
+  - Others: Logs a warning (not implemented).
+- Uses transactions to ensure safe operations.
+
+#### dbflush command example
+
+```bash
+# Flush all models with confirmation
+python manage.py dbflush
+
+# Flush a specific model (User) with confirmation
+python manage.py dbflush --model User
+
+# Flush all models without prompt
+python manage.py dbflush --yes
+```
+
+#### Output
+
+```bash
+Flushed 10 records from model "User" and reset IDs.
+```
+
+or
+
+```bash
+Flushed 120 records from all models and reset IDs.
+```
+
+### 3. EnvBaseSettings
+
+```python
+from djresttoolkit.envconfig import EnvBaseSettings
+```
+
+#### `EnvBaseSettings`
+
+A **base settings class** for managing application configuration using:
+
+- YAML files (default `.environ.yaml`)
+- Environment variables (default `.env`)
+
+Supports **nested configuration** using double underscores (`__`) in environment variable names.
+
+#### Class Attributes
+
+| Attribute      | Type                 | Default         | Description                                                        |
+| -------------- | -------------------- | --------------- | ------------------------------------------------------------------ |
+| `env_file`     | `str`                | `.env`          | Environment variable file path.                                    |
+| `yaml_file`    | `str`                | `.environ.yaml` | YAML configuration file path.                                      |
+| `model_config` | `SettingsConfigDict` | —               | Pydantic settings configuration (file encoding, nested delimiter). |
+
+#### Methods
+
+#### `load(cls, *, env_file: str | None = None, ymal_file: str | None = None, warning: bool = True) -> EnvBaseSettings`
+
+Loads configuration from **YAML first**, then overrides with **environment variables**.
+
+#### Parameters
+
+- `env_file` — Optional custom `.env` file path.
+- `ymal_file` — Optional custom YAML file path.
+- `warning` — Emit a warning if YAML file is missing (default `True`).
+
+#### Returns
+
+- Instance of `EnvBaseSettings` (or subclass) with loaded configuration.
+
+#### Raises
+
+- `UserWarning` if YAML file not found and `warning=True`.
+
+### Usage Example
+
+```python
+from djresttoolkit.envconfig import EnvBaseSettings
+
+class EnvSettings(EnvBaseSettings):
+    debug: bool = False
+    database_url: str
+
+# Load settings
+settings = EnvSettings.load(warning=False)
+
+print(settings.debug)
+print(settings.database_url)
+```
+
+#### Features
+
+- Prioritizes `.env` variables over YAML.
+- Supports nested keys: `DATABASE__HOST` → `settings.database.host`.
+- Designed to be subclassed for project-specific settings.
+
+### 4. EmailSender
 
 ```python
 from djresttoolkit.mail import EmailSender, EmailContent, EmailTemplate
@@ -157,7 +284,7 @@ Send templated emails.
 EmailSender(email_content: EmailContent | EmailContentDict)
 ```
 
-#### Methods
+#### EmailSender Methods
 
 ```python
 send(to: list[str], exceptions: bool = False) -> bool
@@ -190,7 +317,7 @@ EmailSender(content).send(to=["user@example.com"])
 
 - `text`, `html` — template file paths
 
-### 3. Custom DRF Exception Handler
+### 5. Custom DRF Exception Handler
 
 ```python
 from djresttoolkit.views import exception_handler
@@ -204,12 +331,12 @@ A DRF exception handler that:
 - Adds throttling support (defaults to `AnonRateThrottle`).
 - Returns **429 Too Many Requests** with `retry_after` if throttle limit is exceeded.
 
-#### Parameters
+#### Exception Handler Parameters
 
 - `exc`: Exception object.
 - `context`: DRF context dictionary containing `"request"` and `"view"`.
 
-#### Returns
+#### Returns Type of Exception Handler
 
 - `Response` — DRF Response object (with throttling info if applicable), or `None`.
 
@@ -230,7 +357,7 @@ REST_FRAMEWORK = {
 - Tracks requests in cache and calculates `retry_after`.
 - Cleans expired timestamps automatically.
 
-### 4. Response Time Middleware
+### 6. Response Time Middleware
 
 ```python
 from djresttoolkit.middlewares import ResponseTimeMiddleware
@@ -248,7 +375,7 @@ ResponseTimeMiddleware(get_response: Callable[[HttpRequest], HttpResponse])
 
 - `get_response`: The next middleware or view callable.
 
-#### Usage
+#### Response Time Middleware Usage
 
 Add it to your Django `MIDDLEWARE` in `settings.py`:
 
@@ -277,7 +404,7 @@ X-Response-Time: 0.01234 seconds
 INFO: Request processed in 0.01234 seconds
 ```
 
-### 5. Throttle Utilities
+### 7. Throttle Utilities
 
 #### `ThrottleInfoJSONRenderer`
 

{djresttoolkit-0.6.2.dist-info → djresttoolkit-0.8.0.dist-info}/RECORD

@@ -1,20 +1,24 @@
 LICENSE,sha256=8-oZM3yuuTRjySMbVKX9YXYA7Y4M_KhQNBYXPFjeWUo,1074
-README.md,sha256=q-R5E_ZZPZt6vLW4TL-HA8M601_vQrI53IZ-dIAgD9E,6983
+README.md,sha256=9S8AiRr4U5uA-YrgtQbUZiYpbg4rT_UejlDGhJ2WYzM,10558
 demo/staticfiles/admin/img/LICENSE,sha256=0RT6_zSIwWwxmzI13EH5AjnT1j2YU3MwM9j3U19cAAQ,1081
 src/djresttoolkit/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 src/djresttoolkit/admin.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 src/djresttoolkit/apps.py,sha256=nKb5GUIEhAB3IL3lTmEXNc5XuvvaZupH-1CCuYKFrEQ,158
 src/djresttoolkit/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 src/djresttoolkit/dbseed/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-src/djresttoolkit/dbseed/models/__init__.py,sha256=B2WJqw2ncqf5nNhkM3XDFx13oNg2ccsULUQkFAjOyrY,105
+src/djresttoolkit/dbseed/models/__init__.py,sha256=uuynQIcfVqEaZN9hF_caI24zm8az23JdXLDrv7xOKTQ,180
+src/djresttoolkit/dbseed/models/_choice_field.py,sha256=T7LAzbyXqlYp2mtCAKL8E1Da_MEh9RzgLZrFJ7fa4gM,446
 src/djresttoolkit/dbseed/models/_gen.py,sha256=qBPQaLvh1rcEam0YmE4JBJqpa-Vv5IFlIIagkEMHDVw,206
 src/djresttoolkit/dbseed/models/_seed_model.py,sha256=0cmbi0VNKjmJbwhjeCFsvb3iKYjok6TJOk6Y2MF_3N4,2443
+src/djresttoolkit/envconfig/__init__.py,sha256=PcLaPaVfQmz3-4m6SwoOQF2W4U7F0agBGJ4Qjqbcyfw,74
+src/djresttoolkit/envconfig/_env_settings.py,sha256=cKGDLo7WU0wzQGkWSwgrUxGCb0sT111JUt4HGlDVSxo,3221
 src/djresttoolkit/mail/__init__.py,sha256=tB9SdMlhfWQ640q4aobZ0H1c7fTWalpDL2I-onkr2VI,268
 src/djresttoolkit/mail/_email_sender.py,sha256=bPMqgD5HibJcOZgO6xxHOhdK9HEhnGNC6BoMPpo-h7k,3096
 src/djresttoolkit/mail/_models.py,sha256=of5KsLGvsN2OWgDYgdtLEijulg817TXgsLKuUdsnDQc,1447
 src/djresttoolkit/mail/_types.py,sha256=zf6CcXR1ei_UmZ1nLAJa378OAJ6ftnBICqEOkzXPNw8,646
 src/djresttoolkit/management/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 src/djresttoolkit/management/commands/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+src/djresttoolkit/management/commands/dbflush.py,sha256=v7rXcuNQDiXLV7p0xzKMNXVeWAN7QKmq2qQQAbhHEEw,3423
 src/djresttoolkit/management/commands/dbseed.py,sha256=sY87TvvzECG7aQXxsDjr45VzCzJpofLacBK49EYFK8M,4225
 src/djresttoolkit/middlewares/__init__.py,sha256=GZHU3Yy4xXoEi62tHn0UJNxN6XgGM2_HES8Bt5AS5Lk,100
 src/djresttoolkit/middlewares/_response_time_middleware.py,sha256=1wCwdkW5Ng6HJo8zx0F7ylms84OGP-1K0kbyG6Vacuk,908
@@ -27,8 +31,8 @@ src/djresttoolkit/throttling/_throttle_inspector.py,sha256=Kss6ZxKy-EXq9UGaGprGD
 src/djresttoolkit/views/__init__.py,sha256=XrxBrs6sH4HmUzp41omcmy_y94pSaXAVn01ttQ022-4,76
 src/djresttoolkit/views/_exceptions/__init__.py,sha256=DrCUxuPNyBR4WhzNutn5HDxLa--q51ykIxSG7_bFsOI,83
 src/djresttoolkit/views/_exceptions/_exception_handler.py,sha256=_o7If47bzWLl57LeSXSWsIDsJGo2RIpwYAwNQ-hsHVY,2839
-djresttoolkit-0.6.2.dist-info/METADATA,sha256=oeLct3kUn0oN2t0SYWXTp-XYRURhNaAcHi7srpU9mE4,9922
-djresttoolkit-0.6.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-djresttoolkit-0.6.2.dist-info/entry_points.txt,sha256=YMhfTF-7mYppO8QqqWnvR_hyMWvoYxD6XI94_ViFu3k,60
-djresttoolkit-0.6.2.dist-info/licenses/LICENSE,sha256=8-oZM3yuuTRjySMbVKX9YXYA7Y4M_KhQNBYXPFjeWUo,1074
-djresttoolkit-0.6.2.dist-info/RECORD,,
+djresttoolkit-0.8.0.dist-info/METADATA,sha256=qCwxDQnmpbQygUL7gytkkO826sOs9Zm8Xy_NXmlLvQY,13567
+djresttoolkit-0.8.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+djresttoolkit-0.8.0.dist-info/entry_points.txt,sha256=YMhfTF-7mYppO8QqqWnvR_hyMWvoYxD6XI94_ViFu3k,60
+djresttoolkit-0.8.0.dist-info/licenses/LICENSE,sha256=8-oZM3yuuTRjySMbVKX9YXYA7Y4M_KhQNBYXPFjeWUo,1074
+djresttoolkit-0.8.0.dist-info/RECORD,,

src/djresttoolkit/dbseed/models/__init__.py

@@ -1,4 +1,10 @@
 from ._gen import Field, Gen
 from ._seed_model import SeedModel
+from ._choice_field import choice_field
 
-__all__ = ["SeedModel", "Gen", "Field"]
+__all__ = [
+    "SeedModel",
+    "Gen",
+    "Field",
+    "choice_field",
+]

src/djresttoolkit/dbseed/models/_choice_field.py

@@ -0,0 +1,17 @@
+import random
+
+from pydantic import Field
+
+
+def choice_field[T](choices: list[T]) -> T:
+    """
+    Creates a field with a default value randomly selected from the provided choices.
+    Args:
+        choices (list[T]): A list of possible values for the field.
+    Returns:
+        T: A field with a default value randomly chosen from the given choices.
+    """
+
+    return Field(
+        default_factory=lambda: random.choice(seq=choices),
+    )

src/djresttoolkit/envconfig/__init__.py

@@ -0,0 +1,3 @@
+from ._env_settings import EnvBaseSettings
+
+__all__ = ["EnvBaseSettings"]

src/djresttoolkit/envconfig/_env_settings.py

@@ -0,0 +1,84 @@
+import warnings
+from pathlib import Path
+from typing import Any, ClassVar
+
+import yaml
+from pydantic_settings import BaseSettings, SettingsConfigDict
+
+
+class EnvBaseSettings(BaseSettings):
+    """ "
+    EnvBaseSettings is a base settings class for managing application configuration
+    using both YAML files and environment variables.
+    This class is designed to load configuration values from a YAML file first,
+    and then override those values with environment variables if present. It supports
+    nested configuration using a double underscore (`__`) as the delimiter in
+    environment variable names, allowing for hierarchical settings.
+
+    Class Attributes:
+        env_file (str): The default filename for the environment variables file (default: ".env").
+        yaml_file (str): The default filename for the YAML configuration file (default: ".environ.yaml").
+        model_config (SettingsConfigDict): Configuration for environment variable parsing, including file encoding and nested delimiter.
+
+    Methods:
+        load(cls, *, env_file: str | None = None, ymal_file: str | None = None, warning: bool = True) -> "EnvBaseSettings":
+            Loads configuration from a YAML file (if it exists), then overrides with environment variables.
+            - env_file: Optional custom path to the .env file.
+            - ymal_file: Optional custom path to the YAML file.
+            - warning: If True, emits a warning if the YAML file is not found.
+            Returns an instance of EnvBaseSettings with the loaded configuration.
+
+    Usage:
+        - Define your settings as subclasses of EnvBaseSettings.
+        - Call `YourSettingsClass.load()` to load configuration from files and environment variables.
+        - Supports nested configuration via double underscore in environment variable names (e.g., `DATABASE__HOST`).
+
+    Raises:
+        - UserWarning: If the YAML file is not found and `warning` is True.
+
+    Example:
+    ```python
+        from djresttoolkit.envconfig import EnvBaseSettings
+        
+        class EnvSettings(EnvBaseSettings):
+            debug: bool = False
+            database_url: str
+            
+        settings = EnvSettings.load(warning=False)
+    ```
+
+    """
+
+    env_file: ClassVar[str] = ".env"
+    yaml_file: ClassVar[str] = ".environ.yaml"
+
+    model_config = SettingsConfigDict(
+        env_file=env_file,
+        env_file_encoding="utf-8",
+        env_nested_delimiter="__",
+    )
+
+    @classmethod
+    def load(
+        cls,
+        *,
+        env_file: str | None = None,
+        ymal_file: str | None = None,
+        warning: bool = True,
+    ) -> "EnvBaseSettings":
+        """Load from YAML first, then override with .env."""
+        if env_file:
+            cls.env_file = env_file
+        if ymal_file:
+            cls.yaml_file = ymal_file
+
+        config_file = Path(cls.yaml_file)
+        yaml_data: dict[str, Any] = {}
+        if config_file.exists():
+            with config_file.open("r") as f:
+                yaml_data = yaml.safe_load(f) or {}
+        elif warning:
+            msg: str = f"Config file {config_file} not found, using only env vars."
+            warnings.warn(msg, UserWarning, stacklevel=1)
+
+        return cls(**yaml_data)

src/djresttoolkit/management/commands/dbflush.py

@@ -0,0 +1,94 @@
+from typing import Any
+
+from django.apps import apps
+from django.core.management.base import BaseCommand, CommandError, CommandParser
+from django.db import connection, transaction
+
+
+class Command(BaseCommand):
+    help = "Flush database records for all models or a specific model"
+
+    def add_arguments(self, parser: CommandParser) -> None:
+        parser.add_argument(
+            "--model",
+            type=str,
+            help="Specify the model name to flush records (case-sensitive, e.g., User)",
+        )
+        parser.add_argument(
+            "--yes",
+            action="store_true",
+            help="Confirm flush without prompt",
+        )
+
+    def handle(self, *args: Any, **options: Any) -> None:
+        """Handle the flushing of database records and reset IDs."""
+
+        model_name = options.get("model")
+        confirm = options.get("yes", False)
+
+        if not confirm:
+            prompt = "Are you sure you want to flush "
+            prompt += f"the model '{model_name}'" if model_name else "all models"
+            prompt += "? [y/n]: "
+            user_input = input(prompt)
+            if user_input.lower() not in ["y", "yes"]:
+                self.stdout.write(self.style.WARNING("Operation cancelled."))
+                return
+
+        def reset_sequence(table_name: str) -> None:
+            """Reset auto-increment ID sequence for supported databases."""
+            with connection.cursor() as cursor:
+                if connection.vendor == "postgresql":
+                    cursor.execute(
+                        f'ALTER SEQUENCE "{table_name}_id_seq" RESTART WITH 1;'
+                    )
+                elif connection.vendor == "sqlite":
+                    cursor.execute(
+                        f"DELETE FROM sqlite_sequence WHERE name='{table_name}';"
+                    )
+                else:
+                    self.stdout.write(
+                        self.style.WARNING(
+                            f"ID reset not implemented for {connection.vendor}."
+                        )
+                    )
+
+        if model_name:
+            # Find the model automatically across all apps
+            model = None
+            for m in apps.get_models():
+                if m.__name__ == model_name:
+                    model = m
+                    break
+
+            if not model:
+                raise CommandError(
+                    f'Model "{model_name}" not found in any installed app.'
+                )
+
+            table_name = model._meta.db_table
+            with transaction.atomic():
+                deleted_count, _ = model.objects.all().delete()
+                reset_sequence(table_name)
+
+            self.stdout.write(
+                self.style.SUCCESS(
+                    f'Flushed {deleted_count} records from model "{model_name}" and reset IDs.'
+                )
+            )
+
+        else:
+            # Flush all models
+            total_deleted = 0
+            with transaction.atomic():
+                for model in apps.get_models():
+                    table_name = model._meta.db_table
+                    deleted_count, _ = model.objects.all().delete()
+                    total_deleted += deleted_count
+                    reset_sequence(table_name)
+
+            self.stdout.write(
+                self.style.SUCCESS(
+                    f"Flushed {total_deleted} records from all models and reset IDs."
+                )
+            )