Support binary operators

Author: mrkn

lib/pycall/pyobject_wrapper.rb

@@ -13,17 +13,24 @@ def self.extend_object(obj)
     end
 
     OPERATOR_METHOD_NAMES = {
-      :+ => :__add__,
-      :- => :__sub__,
-      :* => :__mul__,
-      :/ => :__truediv__
+      :+  => :__add__,
+      :-  => :__sub__,
+      :*  => :__mul__,
+      :/  => :__truediv__,
+      :%  => :__mod__,
+      :** => :__pow__,
+      :<< => :__lshift__,
+      :>> => :__rshift__,
+      :&  => :__and__,
+      :^  => :__xor__,
+      :|  => :__or__
     }.freeze
 
     def method_missing(name, *args)
       name_str = name.to_s if name.kind_of?(Symbol)
       name_str.chop! if name_str.end_with?('=')
       case name
-      when :+, :-, :*, :/
+      when *OPERATOR_METHOD_NAMES.keys
         op_name = OPERATOR_METHOD_NAMES[name]
         if LibPython::Helpers.hasattr?(__pyptr__, op_name)
           LibPython::Helpers.define_wrapper_method(self, op_name)
@@ -103,6 +110,34 @@ def *(other)
       def /(other)
         other.__rtruediv__(self.obj)
       end
+
+      def %(other)
+        other.__rmod__(self.obj)
+      end
+
+      def **(other)
+        other.__rpow__(self.obj)
+      end
+
+      def <<(other)
+        other.__rlshift__(self.obj)
+      end
+
+      def >>(other)
+        other.__rrshift__(self.obj)
+      end
+
+      def &(other)
+        other.__rand__(self.obj)
+      end
+
+      def ^(other)
+        other.__rxor__(self.obj)
+      end
+
+      def |(other)
+        other.__ror__(self.obj)
+      end
     end
 
     def coerce(other)

spec/pycall/pyobject_wrapper_spec.rb

@@ -88,6 +88,34 @@ module PyCall
       context 'when the name is :/' do
         it 'delegates to :__truediv__'
       end
+
+      context 'when the name is :%' do
+        it 'delegates to :__mod__'
+      end
+
+      context 'when the name is :**' do
+        it 'delegates to :__pow__'
+      end
+
+      context 'when the name is :<<' do
+        it 'delegates to :__lshift__'
+      end
+
+      context 'when the name is :>>' do
+        it 'delegates to :__rshift__'
+      end
+
+      context 'when the name is :&' do
+        it 'delegates to :__and__'
+      end
+
+      context 'when the name is :^' do
+        it 'delegates to :__xor__'
+      end
+
+      context 'when the name is :|' do
+        it 'delegates to :__or__'
+      end
     end
 
     describe '#==' do