Add initial version

Author: codingjoe

.travis.yml

@@ -0,0 +1,29 @@
+language: python
+sudo: false
+cache:
+- pip
+python:
+- '3.4'
+- '3.5'
+- '3.6'
+- '3.7-dev'
+matrix:
+  fast_finish: true
+  allow_failures:
+  - python: '3.7-dev'
+install:
+- pip install -r requirements-dev.txt
+- pip install -U codecov
+script:
+- coverage run --source yeelib -m pytest tests
+after_success:
+- codecov
+deploy:
+  provider: pypi
+  user: codingjoe
+  password:
+    secure: jn763GkmO/nM7TJ9sekgBbFRMI1Z8XijY4ej1ztcJgIwWvZlAWga02sBEnjyFzeqnNmYWAnM50F0oLAvPxmhwIFEUGby2h2Ppy3RhyyNsYi3nQiFKC9/rr0oMJG3xtqyGIJx35pyMM8IzrwQ/tDAcdBRl4HgxnDTLAjgf6Wb1mOhwvQ+/+AZqVqHS0IdXM80K/SgO6LsgL1HgBHdJ7m9i/FUmP5RfDXAPIddd5n81Y73y0ylqsHFgSh8/LtAEcmlJEQIkZDp3jJYM5vxzV4h0LQ6vU7Y1p9RnC6cKEQd+d70iAr+p9apwlwRshaYGy8QVdSbt1vPzpx3mg1Jj1+JpaTmjkC6Ki3XGwcLtARc4MeARnrtB8o0R+QP3KnbfxxQ1SFmvfglZo2HXss6PBB2r6XhAqz7n3zwa7FDqkPb01sytXumnIjckD1E+68X4bR31i627wtIJ5DzTfJYbXAcLbwIPcvnwtEqtmP5IVZJW+xl7ImuyTeslDMx9lT56r2SuGeXZ/B1FRuCRa0hkWasY2DRRo+xrlfleAoMCrLTGKewnNtu9R6r8F1nPUIboLBpirqYStgD86La4uFJai8yer92zNnAue1q7gqj6hAEzoBITGxvDMiy1huWnvnVXP2R9AU/G+oFIGQgbBrXwOVDqeTHla9MGfID3j9rWSi1oNs=
+  on:
+    tags: true
+    distributions: sdist bdist_wheel
+    repo: codingjoe/yeelib

README.md

@@ -0,0 +1,93 @@
+======
+yeelib
+======
+
+**Python library for Xiaomi Mi Yeelight.**
+
+``yeelib`` is an ``asyncio`` based Python library to control Yeelights.
+It has no external dependencies and is blazing fast!
+
+
+Getting Started
+---------------
+
+Install
+
+.. code:: shell
+
+    pip install yeelib
+
+Example:
+
+.. code:: python
+
+    import asyncio
+
+    from yeelib import search_bulbs
+
+
+    @asyncio.coroutine
+    def turn_all_lights_on(bulbs):
+        while True:
+            print(bulbs)
+            for b in bulbs.values():
+                asyncio.Task(b.send_command("set_power",
+                                            ["off", "sudden", 40]))
+            yield from asyncio.sleep(10)
+
+
+    def main():
+        loop = asyncio.get_event_loop()
+        with search_bulbs() as bulbs:
+            loop.create_task(turn_all_lights_on(bulbs))
+            try:
+                loop.run_forever()
+            except KeyboardInterrupt:
+                loop.stop()
+
+
+    if __name__ == '__main__':
+        main()
+
+The script above will turn all off every 10 seconds.
+The following script does the same thing, but note how you can define the bulbs
+prior calling the ``search_bulbs`` context manager. This works do to the fact
+that dictionaries are mutable in Python.
+
+.. code:: python
+
+    import asyncio
+
+    from yeelib import search_bulbs
+
+
+    @asyncio.coroutine
+    def turn_all_lights_on(bulbs):
+        while True:
+            print(bulbs)
+            for b in bulbs.values():
+                asyncio.Task(b.send_command("set_power",
+                                            ["off", "sudden", 40]))
+            yield from asyncio.sleep(10)
+
+
+    def main():
+        bulbs = {}
+        asyncio.Task(turn_all_lights_on(bulbs))
+        loop = asyncio.get_event_loop()
+        try:
+            with search_bulbs(bulbs):
+                loop.run_forever()
+        except KeyboardInterrupt:
+            loop.stop()
+
+
+    if __name__ == '__main__':
+        main()
+
+
+Specifications
+--------------
+
+For more information check out the Yeelight developer documentation.
+http://www.yeelight.com/download/Yeelight_Inter-Operation_Spec.pdf

README.rst

@@ -0,0 +1,4 @@
+-e .
+pytest
+coverage
+mocket

requirements-dev.txt

@@ -0,0 +1,27 @@
+[metadata]
+name = yeelib
+author = Johannes Hoppe
+author-email = info@johanneshoppe.com
+summary = Pyhton library for Xioami Mi Yeelight.
+description-file = README.rst
+description-content-type = text/x-rst; charset=UTF-8
+home-page = https://github.com/codingjoe/yeelib
+license = Apache-2
+classifier =
+    Development Status :: 4 - Beta
+    Environment :: Console
+    Intended Audience :: Developers
+    Intended Audience :: Information Technology
+    License :: OSI Approved :: Apache Software License
+    Operating System :: OS Independent
+    Programming Language :: Python
+keywords =
+    light
+    yeelight
+
+[files]
+packages =
+    yeelib
+
+[pydocstyle]
+add_ignore = D1

setup.cfg

@@ -0,0 +1,7 @@
+#!/usr/bin/env python
+from setuptools import setup
+
+setup(
+    setup_requires=['pbr'],
+    pbr=True,
+)

setup.py

@@ -0,0 +1,45 @@
+import asyncio
+
+from mocket import Mocket, MocketEntry, mocketize
+
+from yeelib import Bulb
+
+
+class TestBulb:
+    bulb_addr = ('192.168.1.239', 55443)
+
+    @mocketize
+    def test_send_command(self):
+        Mocket.register(MocketEntry(self.bulb_addr, [b'{"id":1, "result":["ok"]}\r\n']))
+        with Bulb(*self.bulb_addr) as b:
+            loop = asyncio.get_event_loop()
+            response = loop.run_until_complete(b.send_command('set_ct_abx', [3500, 'smooth', 500]))
+
+        assert response == {'id': 1, 'result': ['ok']}
+
+    def test_repr(self):
+        with Bulb(*self.bulb_addr) as b:
+            assert repr(b) == '<Bulb: 192.168.1.239>'
+
+    def test_kwargs(self):
+        kwargs = {
+            'id': '0x000000000015243f',
+            'model': 'color',
+            'fw_ver': 18,
+            'support': 'get_prop set_default set_power toggle set_bright start_cf stop_cf set_scene'
+                       'cron_add cron_get cron_del set_ct_abx set_rgb',
+            'power': 'on',
+            'bright': 100,
+            'color_mode': 2,
+            'ct': 4000,
+            'rgb': 16711680,
+            'hue': 100,
+            'sat': 35,
+            'name': 'my_bulb',
+        }
+
+        with Bulb(*self.bulb_addr, **kwargs) as b:
+            assert b.fw_ver == 18
+            assert b.support == ['get_prop', 'set_default', 'set_power', 'toggle', 'set_bright',
+                                 'start_cf', 'stop_cf', 'set_scenecron_add', 'cron_get',
+                                 'cron_del', 'set_ct_abx', 'set_rgb']

tests/__init__.py

@@ -0,0 +1,91 @@
+import asyncio
+
+import pytest
+
+from yeelib.discover import YeelightProtocol, search_bulbs
+from yeelib.exceptions import YeelightError
+
+
+notify = b"""NOTIFY * HTTP/1.1
+Host: 239.255.255.250:1982
+Cache-Control: max-age=3600
+Location: yeelight://192.168.1.239:55443
+NTS: ssdp:alive
+Server: POSIX, UPnP/1.0 YGLC/1
+id: 0x000000000015243f
+model: color
+fw_ver: 18
+support: get_prop set_default set_power toggle set_bright start_cf stop_cf set_scene
+cron_add cron_get cron_del set_ct_abx set_rgb
+power: on
+bright: 100
+color_mode: 2
+ct: 4000
+rgb: 16711680
+hue: 100
+sat: 35
+name: my_bulb"""
+
+mcast = b"""HTTP/1.1 200 OK
+Cache-Control: max-age=3600
+Date:
+Ext:
+Location: yeelight://192.168.1.239:55443
+Server: POSIX UPnP/1.0 YGLC/1
+id: 0x000000000015243f
+model: color
+fw_ver: 18
+support: get_prop set_default set_power toggle set_bright start_cf stop_cf set_scene
+cron_add cron_get cron_del set_ct_abx set_rgb
+power: on
+bright: 100
+color_mode: 2
+ct: 4000
+rgb: 16711680
+hue: 100
+sat: 35
+name: my_bulb"""
+
+wrong_location = b"""HTTP/1.1 200 OK
+Cache-Control: max-age=3600
+Date:
+Ext:
+Location: yeelight://not.an.ip:55443
+Server: POSIX UPnP/1.0 YGLC/1"""
+
+
+class TestYeelightProtocoll:
+    def test_notify(self, ):
+        bulbs = {}
+        p = YeelightProtocol(bulbs=bulbs)
+        p.datagram_received(data=notify, addr=('192.168.1.239', 1982))
+        assert len(bulbs) == 1
+        assert bulbs['0x000000000015243f'].ip == '192.168.1.239'
+
+    def test_mcast(self, ):
+        bulbs = {}
+        p = YeelightProtocol(bulbs=bulbs)
+        p.datagram_received(data=mcast, addr=('192.168.1.239', 1982))
+        assert len(bulbs) == 1
+        assert bulbs['0x000000000015243f'].ip == '192.168.1.239'
+
+    def test_duplicate(self):
+        bulbs = {}
+        p = YeelightProtocol(bulbs=bulbs)
+        p.datagram_received(data=notify, addr=('192.168.1.239', 1982))
+        p.datagram_received(data=notify, addr=('192.168.1.239', 1982))
+        assert len(bulbs) == 1
+        assert bulbs['0x000000000015243f'].ip == '192.168.1.239'
+
+    def test_wrong_location(self):
+        bulbs = {}
+        p = YeelightProtocol(bulbs=bulbs)
+        with pytest.raises(YeelightError) as e:
+            p.datagram_received(data=wrong_location, addr=('192.168.1.239', 1982))
+        assert 'Location does not match: yeelight://not.an.ip:55443' in str(e)
+
+
+def test_search_bulbs():
+    loop = asyncio.get_event_loop()
+    with search_bulbs():
+        loop.run_until_complete(asyncio.sleep(1))

tests/test_bulbs.py

@@ -0,0 +1,2 @@
+from .discover import *  # NoQA
+from .bulbs import *  # NoQA