gevent friendly intofy_simple

ginotify_simple.py

"""
A gevent friendly version of inotify_simple

Either instanciate INotify objects directly from `ginotify_simple.INotify`:

```python
from ginotify_simple import INotify, flags

ify1 = INotify()

watch = flags.CREATE | flags.DELETE | flags.MODIFY | flags.DELETE_SELF
ify1.add_watch('/tmp', watch)
```

...or instanciate `inotify_simple.INotify` objects directly by guarding them with
`ginotify_simple.ensure` context manager:

```python
from ginotify_simple import ensure
from inotify_simple import INotify

with ensure():
    ify1 = INotify()

watch = flags.CREATE | flags.DELETE | flags.MODIFY | flags.DELETE_SELF
ify1.add_watch('/tmp', watch)

```
"""

import contextlib

import gevent.select
import inotify_simple

from inotify_simple import flags, masks, Event


@contextlib.contextmanager
def ensure():
    select = inotify_simple.inotify_simple.select
    inotify_simple.inotify_simple.select = gevent.select
    yield
    inotify_simple.inotify_simple.select = select


class INotify(inotify_simple.INotify):

    def __init__(self, *args, **kwargs):
        with ensure():
            super(INotify, self).__init__(*args, **kwargs)