-
Notifications
You must be signed in to change notification settings - Fork 50
Expand file tree
/
Copy pathtype.py
More file actions
154 lines (119 loc) · 3.54 KB
/
type.py
File metadata and controls
154 lines (119 loc) · 3.54 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
"""
Copyright (c) Microsoft Corporation.
Licensed under the MIT license.
This module contains type objects and constructors for the mssql_python package.
"""
import datetime
import time
# Type Objects
class STRING(str):
"""
This type object is used to describe columns in a database that are string-based (e.g. CHAR).
"""
def __new__(cls):
return str.__new__(cls, "")
class BINARY(bytearray):
"""
This type object is used to describe (long)
binary columns in a database (e.g. LONG, RAW, BLOBs).
"""
def __new__(cls):
return bytearray.__new__(cls)
class NUMBER(float):
"""
This type object is used to describe numeric columns in a database.
"""
def __new__(cls):
return float.__new__(cls, 0.0)
class DATETIME(datetime.datetime):
"""
This type object is used to describe date/time columns in a database.
"""
def __new__(
cls,
year: int = 1,
month: int = 1,
day: int = 1,
hour: int = 0,
minute: int = 0,
second: int = 0,
microsecond: int = 0,
tzinfo=None,
*,
fold: int = 0,
):
return datetime.datetime.__new__(
cls, year, month, day, hour, minute, second, microsecond, tzinfo, fold=fold
)
class ROWID(int):
"""
This type object is used to describe the "Row ID" column in a database.
"""
def __new__(cls):
return int.__new__(cls, 0)
# Type Constructors
def Date(year: int, month: int, day: int) -> datetime.date:
"""
Generates a date object.
"""
return datetime.date(year, month, day)
def Time(hour: int, minute: int, second: int) -> datetime.time:
"""
Generates a time object.
"""
return datetime.time(hour, minute, second)
def Timestamp(
year: int,
month: int,
day: int,
hour: int,
minute: int,
second: int,
microsecond: int,
) -> datetime.datetime:
"""
Generates a timestamp object.
"""
return datetime.datetime(year, month, day, hour, minute, second, microsecond)
def DateFromTicks(ticks: int) -> datetime.date:
"""
Generates a date object from ticks.
"""
return datetime.date.fromtimestamp(ticks)
def TimeFromTicks(ticks: int) -> datetime.time:
"""
Generates a time object from ticks.
"""
return datetime.time(*time.localtime(ticks)[3:6])
def TimestampFromTicks(ticks: int) -> datetime.datetime:
"""
Generates a timestamp object from ticks.
"""
return datetime.datetime.fromtimestamp(ticks)
def Binary(value) -> bytes:
"""
Converts a string or bytes to bytes for use with binary database columns.
This function follows the DB-API 2.0 specification.
It accepts only str and bytes/bytearray types to ensure type safety.
Args:
value: A string (str) or bytes-like object (bytes, bytearray)
Returns:
bytes: The input converted to bytes
Raises:
TypeError: If the input type is not supported
Examples:
Binary("hello") # Returns b"hello"
Binary(b"hello") # Returns b"hello"
Binary(bytearray(b"hi")) # Returns b"hi"
"""
if isinstance(value, bytes):
return value
if isinstance(value, bytearray):
return bytes(value)
if isinstance(value, str):
return value.encode("utf-8")
# Raise TypeError for unsupported types to improve type safety
raise TypeError(
f"Cannot convert type {type(value).__name__} to bytes. "
f"Binary() only accepts str, bytes, or bytearray objects."
)