openstack/deb-python-cassandra-driver
Code Issues Proposed changes

Use io.BytesIO for collection serialization

Browse Source
This commit is contained in:
Tyler Hobbs
2014-06-17 15:55:00 -05:00
parent 48582f4476
commit e7187abfb4
1 changed files with 6 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
cassandra/cqltypes.py
View File

@@ -27,8 +27,10 @@ the corresponding CQL or Cassandra type strings.
# for example), these classes would be a good place to tack on # for example), these classes would be a good place to tack on
# .from_cql_literal() and .as_cql_literal() classmethods (or whatever). # .from_cql_literal() and .as_cql_literal() classmethods (or whatever).
from __future__ import absolute_import # to enable import io from stdlib
import calendar import calendar
from decimal import Decimal from decimal import Decimal
import io
import re import re
import socket import socket
import time import time
@@ -560,7 +562,8 @@ class DateType(_CassandraType):
if not _have_warned_about_timestamps: if not _have_warned_about_timestamps:
_have_warned_about_timestamps = True _have_warned_about_timestamps = True
warnings.warn("timestamp columns in Cassandra hold a number of " warnings.warn(
"timestamp columns in Cassandra hold a number of "
"milliseconds since the unix epoch. Currently, when executing " "milliseconds since the unix epoch. Currently, when executing "
"prepared statements, this driver multiplies timestamp " "prepared statements, this driver multiplies timestamp "
"values by 1000 so that the result of time.time() " "values by 1000 so that the result of time.time() "
@@ -666,7 +669,7 @@ class _SimpleParameterizedType(_ParameterizedType):
raise TypeError("Received a string for a type that expects a sequence") raise TypeError("Received a string for a type that expects a sequence")
subtype, = cls.subtypes subtype, = cls.subtypes
buf = six.BytesIO() buf = io.BytesIO()
buf.write(uint16_pack(len(items))) buf.write(uint16_pack(len(items)))
for item in items: for item in items:
itembytes = subtype.to_binary(item) itembytes = subtype.to_binary(item)
@@ -719,7 +722,7 @@ class MapType(_ParameterizedType):
@classmethod @classmethod
def serialize_safe(cls, themap): def serialize_safe(cls, themap):
subkeytype, subvaltype = cls.subtypes subkeytype, subvaltype = cls.subtypes
buf = six.BytesIO() buf = io.BytesIO()
buf.write(uint16_pack(len(themap))) buf.write(uint16_pack(len(themap)))
try: try:
items = six.iteritems(themap) items = six.iteritems(themap)