wolever

# A quick but effective hack for disabling migrations while running Django tests.
# Put this in settings.py below the existing MIGRATION_MODULES definition, if that exists:

if "test" in sys.argv[1:]:
    class DisableMigrations(object):
        def __contains__(self, item):
            return True
        def __getitem__(self, item):
            return "notmigrations"

wolever

import inspect

class ABCMeta(type):
    """ A very simple abstract base class implementation which verifies,
        at compile time, that subclasses implement all required abstract
        methods.

        Methods are considered abstract if their implementation contains the
        string ``raise NotImplementedError``.
        

wolever

"""
A simple watchdog for long running processes which may stall for some reason or
another.

If the main thread hasn't logged progress (by updating
``self.last_progress_time``) in WATCHDOG_HARD_KILL_TIMEOUT, the watchdog
thread will log an error containing the stack trace of all currently running
threads then use ``kill -9`` to kill the main process.

Assumes that a process monitor like supervisor or systemd will then restart

wolever

By shazow and wolever.

Problem

How can N secret santa gift exchange participants agree on a gift-giver -> gift-recipient bijection such that:

wolever

"""
Testcase Helpers (pending a better name) make it easy to compose elements
which require setup/teardown on each testcase.

For example, consider a testcase which needs to clean a Redis database after
each run::

    class TestRedis(TestCase):
        def cleanup_redis(self):
            for key in self.cxn.keys("*"):

wolever

"""
I have F8 in Vim (see pdb.vim) bound to insert this statement.

It will:
- Still work if running tests under nosetests
- Can be disabled if it's accidentally added in a loop by setting "st.off = True"
- The XXX is highlighted in Vim, and BREAK is easy to grep for
"""

from nose.tools import set_trace as st; st.__dict__.get("off") or st() #XXX: BREAK

wolever

import time

import redis


def get_redis_connection():
    return redis.connect()

class TaskDebouncer(object):
    """ A simple Celery task debouncer.

wolever

from django.contribu.auth.backends import ModelBackend
from django.contrib.auth import get_user_model

class UsernameOrEmailBackend(ModelBackend):
    """ An authentication backend which will allow users to login using
        either their username or their email.
        
        In settings.py, add:
        
            AUTHENTICATION_BACKENDS = [

wolever

$ coverage run test.py
...
----------------------------------------------------------------------
Ran 3 tests in 0.000s

OK
$ coverage report
Name                                                                                                Stmts   Miss  Cover
-----------------------------------------------------------------------------------------------------------------------
/Users/wolever/code/sandbox/env/sandbox/lib/python2.7/site-packages/pkg_resources/_vendor/six         444    442     1%

wolever

@contextmanager
def spinlock(key, suffix=":lock", duration=1, spinsleep=0.01):
    """ A simple Redis-based spinlock for situations where pessimistic locking
        makes life simpler, and the chance of collisions is low enough that
        a spinlock is a reasonable tradeoff.

        The lock is held for at most ``duration`` seconds (after which it's
        ignored), and polled every ``spinsleep`` seconds.

        For example::