Skip to content

Commit f623911

Browse files
committed
made naming conventions consistent in alter.py
1 parent 234879f commit f623911

File tree

1 file changed

+10
-10
lines changed

1 file changed

+10
-10
lines changed

piccolo/query/methods/alter.py

Lines changed: 10 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -107,7 +107,7 @@ def querystring(self) -> QueryString:
107107

108108

109109
@dataclass
110-
class Unique(AlterColumnStatement):
110+
class SetUnique(AlterColumnStatement):
111111
__slots__ = ("boolean",)
112112

113113
boolean: bool
@@ -129,7 +129,7 @@ def querystring(self) -> QueryString:
129129

130130

131131
@dataclass
132-
class Null(AlterColumnStatement):
132+
class SetNull(AlterColumnStatement):
133133
__slots__ = ("boolean",)
134134

135135
boolean: bool
@@ -255,13 +255,13 @@ class Alter(Query):
255255
"_drop_default",
256256
"_drop_table",
257257
"_drop",
258-
"_null",
259258
"_rename_columns",
260259
"_rename_table",
261260
"_set_default",
262261
"_set_digits",
263262
"_set_length",
264-
"_unique",
263+
"_set_null",
264+
"_set_unique",
265265
)
266266

267267
def __init__(self, table: t.Type[Table]):
@@ -272,13 +272,13 @@ def __init__(self, table: t.Type[Table]):
272272
self._drop_default: t.List[DropDefault] = []
273273
self._drop_table: t.Optional[DropTable] = None
274274
self._drop: t.List[DropColumn] = []
275-
self._null: t.List[Null] = []
276275
self._rename_columns: t.List[RenameColumn] = []
277276
self._rename_table: t.List[RenameTable] = []
278277
self._set_default: t.List[SetDefault] = []
279278
self._set_digits: t.List[SetDigits] = []
280279
self._set_length: t.List[SetLength] = []
281-
self._unique: t.List[Unique] = []
280+
self._set_null: t.List[SetNull] = []
281+
self._set_unique: t.List[SetUnique] = []
282282

283283
def add_column(self, name: str, column: Column) -> Alter:
284284
"""
@@ -349,7 +349,7 @@ def set_null(
349349
Band.alter().set_null(Band.name, True)
350350
Band.alter().set_null('name', True)
351351
"""
352-
self._null.append(Null(column, boolean))
352+
self._set_null.append(SetNull(column, boolean))
353353
return self
354354

355355
def set_unique(
@@ -359,7 +359,7 @@ def set_unique(
359359
Band.alter().set_unique(Band.name, True)
360360
Band.alter().set_unique('name', True)
361361
"""
362-
self._unique.append(Unique(column, boolean))
362+
self._set_unique.append(SetUnique(column, boolean))
363363
return self
364364

365365
def set_length(self, column: t.Union[str, Varchar], length: int) -> Alter:
@@ -472,8 +472,8 @@ def querystrings(self) -> t.Sequence[QueryString]:
472472
self._rename_table,
473473
self._drop,
474474
self._drop_default,
475-
self._unique,
476-
self._null,
475+
self._set_unique,
476+
self._set_null,
477477
self._set_length,
478478
self._set_default,
479479
self._set_digits,

0 commit comments

Comments
 (0)