mirror of
https://github.com/sqlalchemy/sqlalchemy.git
synced 2026-05-28 11:35:19 -04:00
Mike Bayer
1e278de4cc
Applied on top of a pure run of black -l 79 in I7eda77fed3d8e73df84b3651fd6cfcfe858d4dc9, this set of changes resolves all remaining flake8 conditions for those codes we have enabled in setup.cfg. Included are resolutions for all remaining flake8 issues including shadowed builtins, long lines, import order, unused imports, duplicate imports, and docstring issues. Change-Id: I4f72d3ba1380dd601610ff80b8fb06a2aff8b0fe
65 lines
1.7 KiB
Python
65 lines
1.7 KiB
Python
"""Installs some sample data. Here we have a handful of postal codes for
|
|
a few US/Canadian cities. Then, 100 Person records are installed, each
|
|
with a randomly selected postal code.
|
|
|
|
"""
|
|
import random
|
|
|
|
from .environment import Base
|
|
from .environment import Session
|
|
from .model import Address
|
|
from .model import City
|
|
from .model import Country
|
|
from .model import Person
|
|
from .model import PostalCode
|
|
|
|
|
|
def install():
|
|
Base.metadata.create_all(Session().bind)
|
|
|
|
data = [
|
|
("Chicago", "United States", ("60601", "60602", "60603", "60604")),
|
|
("Montreal", "Canada", ("H2S 3K9", "H2B 1V4", "H7G 2T8")),
|
|
("Edmonton", "Canada", ("T5J 1R9", "T5J 1Z4", "T5H 1P6")),
|
|
(
|
|
"New York",
|
|
"United States",
|
|
("10001", "10002", "10003", "10004", "10005", "10006"),
|
|
),
|
|
(
|
|
"San Francisco",
|
|
"United States",
|
|
("94102", "94103", "94104", "94105", "94107", "94108"),
|
|
),
|
|
]
|
|
|
|
countries = {}
|
|
all_post_codes = []
|
|
for city, country, postcodes in data:
|
|
try:
|
|
country = countries[country]
|
|
except KeyError:
|
|
countries[country] = country = Country(country)
|
|
|
|
city = City(city, country)
|
|
pc = [PostalCode(code, city) for code in postcodes]
|
|
Session.add_all(pc)
|
|
all_post_codes.extend(pc)
|
|
|
|
for i in range(1, 51):
|
|
person = Person(
|
|
"person %.2d" % i,
|
|
Address(
|
|
street="street %.2d" % i,
|
|
postal_code=all_post_codes[
|
|
random.randint(0, len(all_post_codes) - 1)
|
|
],
|
|
),
|
|
)
|
|
Session.add(person)
|
|
|
|
Session.commit()
|
|
|
|
# start the demo fresh
|
|
Session.remove()
|