You're browsing snippets anonymously. Log in or create an account to save and manage your own snippets.

Public Snippets

using language: Python

Showing 7 snippets

Create safe filename string

def safe_filename(s: str, replacement: str = "_") -> str:
    """
    Convert a string to a safe filename by replacing unsafe characters.

Args:
        s (str): The input string to be converted.
        replacement (str): The character to replace unsafe characters with. Default is "_".

Returns:
        str: A safe filename string.
    """

keep_chars = (" ", ".", "_", "-")
    safe_name = "".join(c if c.isalnum() or c in keep_chars else replacement for c in s)
    safe_name = safe_name.strip().strip(".")
    return safe_name

if __name__ == "__main__":
    test_cases_safe_filename = [
        ("example<file>name.txt", "example_file_name.txt"),
        ("unsafe:/\\|?*name.doc", "unsafe______name.doc"),
        ("normal_filename.pdf", "normal_filename.pdf"),
        ("   leading and trailing spaces   ", "leading and trailing spaces"),
        ("file.name.with.many.dots...", "file.name.with.many.dots"),
        ("control\x00char\x1fname", "control_char_name"),
    ]

# Test safe_filename function
    print("Testing safe_filename function:")
    for input_str, expected_output in test_cases_safe_filename:
        result = safe_filename(input_str)
        assert result == expected_output, (
            f"Failed for input '{input_str}': got '{result}', expected '{expected_output}'"
        )

print("\nAll tests passed!")

Python Public

def safe_filename(s: str, replacement: str = "_") -> str:
    """
    Convert a string to a safe filename by replacing unsafe characters.

    Args:
        s (str): The input string to be conver...

By xtream1101 • Created 2025-10-15 14:28 • Updated 2025-10-15 14:35

Django Query Stats per request

# stats.py
import time

from django.db import connection
from loguru import logger

class StatsMiddleware:
    def __init__(self, get_response):
        self.get_response = get_response

def __call__(self, request):
        """Capture stats about each request
        """
        start_time = time.time()

# Needed in upper environments to get the SQL queries stats
        # Otherwise only works when DEBUG="true"
        connection.force_debug_cursor = True

queries_before = len(connection.queries)
        response = self.get_response(request)
        queries_after = len(connection.queries)

duration = time.time() - start_time

num_queries = queries_after - queries_before

try:
            total_query_time = sum(float(query["time"]) for query in connection.queries)
            logger.info(
                f"Time taken: {duration * 1_000:.2f} ms, "
                f"SQL queries executed: {num_queries}, "
                f"Total query time: {total_query_time * 1_000:.2f} ms",
                num_queries=num_queries,
                duration=duration,
                total_query_time_ms=int(total_query_time * 1_000),
                request_body=request.body.decode("utf-8"),
                request_method=request.method,
                request_path=request.path,
            )

# Add some stats to the header so that we can see them in the clients
            response["X-Response-Time-ms"] = int(duration * 1_000)
            response["X-Total-Queries-Time-ms"] = int(total_query_time * 1_000)
            response["X-Total-Queries"] = num_queries

except Exception:
            # Never expect this to have an exception
            # but better to catch and log, rather then breaking the response
            logger.exception("Error logging request stats")

return response

Python Public

# stats.py
import time

from django.db import connection
from loguru import logger


class StatsMiddleware:
    def __init__(self, get_response):
        self.get_response = get_response

 ...

By xtream1101 • Created 2025-05-11 00:03 • Updated 2025-05-11 00:04

Reset default python log handlers

Python Public aws lambda logging

import logging

formatter = logging.Formatter('FOO: %(message)s')

root_logger = logging.getLogger()
root_logger.setLevel(logging.INFO)

# This is the key line that removes the double logging
...

By xtream1101 • Created 2025-03-02 15:43 • Updated 2025-03-04 18:33

Django graphene mutation input validation

import graphene

class ScratchpadMutation(graphene.relay.ClientIDMutation):
    class Input:
        created_by = graphene.String(required=True)
        foobar = graphene.String()

test = graphene.String()
    test2 = graphene.String()

@classmethod
    def validate_created_by(cls, value: str) -> str:
        if "foo" in value:
            raise ValueError("No foo's allowed")
        return value.strip().upper()

@classmethod
    def validate_combinations(cls, input: dict) -> dict:
        if input.get("created_by") == "TEST" and input.get("foobar") == "TEST":
            raise ValueError("Cannot have both created_by and foobar as TEST")

return input

@classmethod
    def validate_inputs(cls, input: dict) -> dict:
        validated_inputs = input.copy()
        for key, value in input.items():
            validation_method = f"validate_{key}"
            # Check if the method exists and run if so
            if hasattr(cls, validation_method):
                method = getattr(cls, validation_method)
                validated_inputs[key] = method(value)

# Run other validations on all the inputs after the pre-validate/cleaning
        validated_inputs = cls.validate_combinations(validated_inputs)

return validated_inputs

@classmethod
    def mutate_and_get_payload(cls, root, info, **input):
        # Need to update since not all fields may have a validation method
        input = cls.validate_inputs(input)
        
        # do somthing with the inputs here
        
        return ScratchpadMutation(
            test=input.get("created_by"), test2=input.get("foobar")
        )

Python Public django graphene graphql validation

import graphene

class ScratchpadMutation(graphene.relay.ClientIDMutation):
    class Input:
        created_by = graphene.String(required=True)
        foobar = graphene.String()

    test = g...

By xtream1101 • Created 2025-02-06 14:13 • Updated 2025-02-06 14:13

Try, try again - Multiple try/except blocks in a flat format

Python Public exceptions

data = {'some_key': 'key value'}
key_data = None
for _ in range(1):
    try:
        key_data = data['someKey']
    except Exception: pass
    else: break  # It worked

    try:
        key_d...

By xtream1101 • Created 2025-02-06 13:35 • Updated 2025-02-06 13:43

Multi key sort on a list of dicts

Python Public sorting

data = [
    {'name': 'Alice', 'age': 25, 'city': 'New York'},
    {'name': 'Bob', 'age': 30, 'city': 'London'},
    {'name': 'Charlie', 'age': 20, 'city': 'New York'}
]

# Sort by 'city' first,...

By xtream1101 • Created 2025-02-06 13:16 • Updated 2025-02-06 13:27

Python command test

Python Public

import sys

if __name__ == "__main__":
	print(f"args passed in: {sys.argv[1:]}")

By xtream1101 • Created 2025-02-06 05:12 • Updated 2025-02-06 05:12

Showing 7 snippets

Create safe filename string

def safe_filename(s: str, replacement: str = "_") -> str:
    """
    Convert a string to a safe filename by replacing unsafe characters.

Args:
        s (str): The input string to be converted.
        replacement (str): The character to replace unsafe characters with. Default is "_".

Returns:
        str: A safe filename string.
    """

keep_chars = (" ", ".", "_", "-")
    safe_name = "".join(c if c.isalnum() or c in keep_chars else replacement for c in s)
    safe_name = safe_name.strip().strip(".")
    return safe_name

print("\nAll tests passed!")

Python Public

Snippet

def safe_filename(s: str, replacement: str = "_") -> str:
    """
    Convert a string to a safe filename by replacing unsafe characters.

    Args:
        s (str): The input string to be converted.
        replacement (str): The character to replace unsafe characters with. Default is "_".

    Returns:
        str: A safe filename string.
    """

    keep_chars = (" ", ".", "_", "-")
    safe_name = "".join(c if c.isalnum() or c in keep_chars else replacement for c in s)
    safe_name = safe_name.strip().strip(".")
    return safe_name


if __name__ == "__main__":
    test_cases_safe_filename = [
        ("example<file>name.txt", "example_file_name.txt"),
        ("unsafe:/\\|?*name.doc", "unsafe______name.doc"),
        ("normal_filename.pdf", "normal_filename.pdf"),
        ("   leading and trailing spaces   ", "leading and trailing spaces"),
        ("file.name.with.many.dots...", "file.name.with.many.dots"),
        ("control\x00char\x1fname", "control_char_name"),
    ]

    # Test safe_filename function
    print("Testing safe_filename function:")
    for input_str, expected_output in test_cases_safe_filename:
        result = safe_filename(input_str)
        assert result == expected_output, (
            f"Failed for input '{input_str}': got '{result}', expected '{expected_output}'"
        )

    print("\nAll tests passed!")

By xtream1101 • Created 2025-10-15 14:28 • Updated 2025-10-15 14:35