feat(py): add copy_rect and bezier2 methods

Also return an object from __exit__ - that's important.

Author: atalii

bindings/py/examples/drawing.py

@@ -0,0 +1,25 @@
+#!/usr/bin/env python3
+
+"""Create a pane and keep it around for a few seconds."""
+
+import ttds_py
+import time
+
+NET_LOC = "100.95.28.105:8080"
+NAME = "unique-name"
+
+# Create a connection to NET_LOC. For testing purposes, I've got this hardcoded
+# to my testing machine's IP, but you'll probably want to make it
+# ttds.tali.network.
+conn = ttds_py.Connection(NET_LOC)
+
+# Create a pane with the given name and set it to a dark blue-ish color.
+with ttds_py.Pane(NAME, ttds_py.Color(12, 12, 34), conn) as pane:
+    # Keep it around for a second. Once sleep is done and the with block's
+    # scope is exited, the pane will be deleted.
+    #
+    # Note that five seconds might not be long enough for your pane to come up
+    # on the display if others are switching theirs in and out.
+    pane.bezier2(0, 0, 200, 200, 500, 500, ttds_py.Color(100, 100, 255))
+    pane.copy_rect(0, 0, 200, 200, 300, 200)
+    time.sleep(5)

bindings/py/ttds_py/__init__.py

@@ -58,14 +58,12 @@ class Pane:
 
     def __enter__(self):
         self._token = self.conn.request("pane", self.name, "create", color=self.color)
+        return self
 
     def __exit__(self, exc_type, exc_val, exc_traceback):
         assert self._token is not None
         self.conn.request("pane", self.name, auth=self._token, method="DELETE")
 
-    # TODO: copy_rect, bezier2 methods.
-    # Also, a testing environment in which I port forward into
-    # ttds.tali.network or the like.
     def rect(self, x: int, y: int, w: int, h: int, color: Color):
         self._draw("rect", x=x, y=y, w=w, h=h, color=color)
 
@@ -75,6 +73,12 @@ def circle(self, x: int, y: int, r: int, color: Color):
     def line(self, x: int, y: int, x2: int, y2: int, color: Color):
         self._draw("line", x=x, y=y, x2=x2, y2=y2, color=color)
 
+    def copy_rect(self, x: int, y: int, w: int, h: int, x2: int, y2: int):
+        self._draw("copy_rect", x=x, y=y, w=w, h=h, x2=x2, y2=y2)
+
+    def bezier2(self, x0, y0, x1, y1, x2, y2, color: Color):
+        self._draw("bezier2", x0=x0, y0=y0, x1=x1, y1=y1, x2=x2, y2=y2, color=color)
+
     def _draw(self, shape, **kwargs):
         assert self._token is not None
         self.conn.request("pane", self.name, shape, auth=self._token, **kwargs)