first commit

Author: whtsky

.travis.yml

@@ -0,0 +1,14 @@
+language: python
+
+python:
+  - 2.6
+  - 2.7
+  - 3.3
+  - 3.4
+  - 3.5
+  - pypy
+
+before_install: pip install Cython
+install: python setup.py install
+script: coverage run --source=bencoder setup.py -q nosetests
+after_success: coveralls

bencoder.pyx

@@ -0,0 +1,133 @@
+#cython: language_level=3
+
+# The contents of this file are subject to the BitTorrent Open Source License
+# Version 1.1 (the License).  You may not copy or use this file, in either
+# source code or executable form, except in compliance with the License.  You
+# may obtain a copy of the License at http://www.bittorrent.com/license/.
+#
+# Software distributed under the License is distributed on an AS IS basis,
+# WITHOUT WARRANTY OF ANY KIND, either express or implied.  See the License
+# for the specific language governing rights and limitations under the
+# License.
+
+# Based on https://github.com/karamanolev/bencode3/blob/master/bencode.py
+
+class BTFailure(Exception):
+    pass
+
+
+def decode_int(bytes x, int f):
+    f += 1
+    new_f = x.index(b'e', f)
+    n = int(x[f:new_f])
+    if x[f] == ord('-'):
+        if x[f + 1] == ord('0'):
+            raise ValueError()
+    elif x[f] == ord('0') and new_f != f + 1:
+        raise ValueError()
+    return n, new_f + 1
+
+
+def decode_string(bytes x, int f):
+    colon = x.index(b':', f)
+    n = int(x[f:colon])
+    if x[f] == ord('0') and colon != f + 1:
+        raise ValueError()
+    colon += 1
+    return x[colon:colon + n], colon + n
+
+
+def decode_list(bytes x, int f):
+    r, f = [], f + 1
+    while x[f] != ord('e'):
+        v, f = decode_func[x[f]](x, f)
+        r.append(v)
+    return r, f + 1
+
+
+def decode_dict(bytes x, int f):
+    r, f = {}, f + 1
+    while x[f] != ord('e'):
+        k, f = decode_string(x, f)
+        r[k], f = decode_func[x[f]](x, f)
+    return r, f + 1
+
+
+decode_func = {
+    ord('l'): decode_list,
+    ord('d'): decode_dict,
+    ord('i'): decode_int,
+    ord('1'): decode_string,
+    ord('2'): decode_string,
+    ord('0'): decode_string,
+    ord('3'): decode_string,
+    ord('4'): decode_string,
+    ord('5'): decode_string,
+    ord('6'): decode_string,
+    ord('7'): decode_string,
+    ord('8'): decode_string,
+    ord('9'): decode_string,
+}
+
+
+def bdecode(bytes x):
+    try:
+        r, l = decode_func[x[0]](x, 0)
+    except (IndexError, KeyError, ValueError):
+        raise BTFailure("not a valid bencoded string")
+    if l != len(x):
+        raise BTFailure("invalid bencoded value (data after valid prefix)")
+    return r
+
+
+def encode_int(int x, list r):
+    r.extend((b'i', str(x).encode(), b'e'))
+
+
+def encode_bool(x, list r):
+    if x:
+        encode_int(1, r)
+    else:
+        encode_int(0, r)
+
+
+def encode_string(x, list r):
+    if isinstance(x, str):
+        x = x.encode()
+    r.extend((str(len(x)).encode(), b':', x))
+
+
+def encode_list(x, list r):
+    r.append(b'l')
+    for i in x:
+        encode_func[type(i)](i, r)
+    r.append(b'e')
+
+
+def encode_dict(dict x, list r):
+    r.append(b'd')
+    item_list = list(x.items())
+    item_list.sort()
+    for k, v in item_list:
+        if isinstance(k, str):
+            k = k.encode()
+        r.extend((str(len(k)).encode(), b':', k))
+        encode_func[type(v)](v, r)
+    r.append(b'e')
+
+
+encode_func = {
+    int: encode_int,
+    bytes: encode_string,
+    str: encode_string,
+    list: encode_list,
+    tuple: encode_list,
+    dict: encode_dict,
+    bool: encode_bool,
+}
+
+
+def bencode(x):
+    r = []
+    encode_func[type(x)](x, r)
+    return b''.join(r)

setup.py

@@ -0,0 +1,40 @@
+import sys
+
+from setuptools import setup
+from setuptools.extension import Extension
+from setuptools.dist import Distribution
+Distribution(dict(setup_requires='Cython'))
+
+try:
+    from Cython.Distutils import build_ext
+except ImportError:
+    print("Could not import Cython.Distutils. Install `cython` and rerun.")
+    sys.exit(1)
+
+ext_modules = [Extension("bencoder", ["bencoder.pyx"])]
+
+setup(
+    name='bencoder.pyx',
+    version='1.0',
+    description='Yet another bencode implementation in Cython',
+    long_description="""""",
+    author='whtsky',
+    author_email='whtsky@gmail.com',
+    url='https://github.com/whtsky/bencoder.pyx',
+    license="BSDv3",
+    platforms=['POSIX', 'Windows'],
+    keywords=['bencoding', 'encode', 'decode', 'bittorrent'],
+    classifiers=[
+        "Environment :: Other Environment",
+        "Intended Audience :: Developers",
+        "Operating System :: OS Independent",
+        "Programming Language :: Python :: 2",
+        "Programming Language :: Python :: 3",
+        "Topic :: Software Development :: Libraries :: Python Modules",
+    ],
+    cmdclass={'build_ext': build_ext},
+    ext_modules=ext_modules,
+    setup_requires=['Cython'],
+    tests_require=['nose'],
+    test_suite='nose.collector',
+)