def handle(self, *args, **options):
categories = json_loader('new_categories')
for category in categories:
new_category = Category(**category)
new_category.save()
print('new_category added')
def handle(self, *args, **options):
categories = loadFromJSON('categories')
Category.objects.all().delete()
for category in categories:
new_category = Category(**category)
new_category.save()
def handle(self, *args, **options):
gender = loadFromJSON('genders')
Gender.objects.all().delete()
for category in gender:
new_category = Gender(**category)
new_category.save()
categories = loadFromJSON('categories')
Category.objects.all().delete()
for category in categories:
new_category = Category(**category)
new_category.save()
menu_category = loadFromJSON('menu_categories')
MenuCategory.objects.all().delete()
for category in menu_category:
new_category = MenuCategory(**category)
new_category.save()
brand = loadFromJSON('brands')
Brand.objects.all().delete()
for category in brand:
new_category = Brand(**category)
new_category.save()
size = loadFromJSON('sizes')
Size.objects.all().delete()
for category in size:
new_category = Size(**category)
new_category.save()
products = loadFromJSON('products')
Product.objects.all().delete()
for product in products:
# заменяем пол на объект:
gender_name = product["gender"]
_gender = Gender.objects.get(name=gender_name)
product["gender"] = _gender
# заменяем категорию на объект:
category_name = product["category"]
_category = Category.objects.get(name=category_name)
product["category"] = _category
# заменяем меню_категорию на объект:
menu_category_name = product["menu_category"]
_menu_category = MenuCategory.objects.get(name=menu_category_name)
product["menu_category"] = _menu_category
# заменяем бренд на объект:
brand_name = product["brand"]
_brand = Brand.objects.get(name=brand_name)
product["brand"] = _brand
# заменяем размер на объект:
size_name = product["size"]
_size = Size.objects.get(name=size_name)
product["size"] = _size
# пихаем все в Product и сохраняем:
new_product = Product(**product)
new_product.save()
def handle(self, *args, **options):
categories = loadFromJSON('categories')
Category.objects.all().delete()
for category in categories:
new_category = Category(**category)
new_category.save()
products = loadFromJSON('products')
Product.objects.all().delete()
for product in products:
category_name = product["category"]
_category = Category.objects.get(name=category_name)
product['category'] = _category
new_product = Product(**product)
new_product.save()
def handle(self, *args, **options):
categories = loadFromJSON('categories')
Category.objects.all().delete()
for category in categories:
new_category = Category(**category)
new_category.save()
products = loadFromJSON('products')
for product in products:
category_name = product["category"]
_category = Category.objects.get(name=category_name)
product['category'] = _category
new_product = Product(**product)
new_product.save()
super_user = ShopUser.objects.create_superuser('admin', '*****@*****.**', '123456', age=28)
def handle(self, *args, **options):
categories = load_json('categories')
Category.objects.all().delete()
for category in categories:
new_category = Category(**category)
new_category.save()
products = load_json('products')
Product.objects.all().delete()
for product in products:
category_name = product["category"]
# Получаем категорию по имени
_category = Category.objects.get(name=category_name)
# Заменяем название категории объектом
product['category'] = _category
new_product = Product(**product)
new_product.save()
def handle(self, *args, **options):
data = load_from_json()
Category.objects.all().delete()
for category in data['categories']:
new_cat = Category(**category)
print(f'Добавлена категория {new_cat}')
new_cat.save()
Brand.objects.all().delete()
for brand in data['brands']:
new_brand = Brand(**brand)
print(f'Добавлен бренд {new_brand}')
new_brand.save()
Product.objects.all().delete()
for product in data['products']:
_category = Category.objects.get(name=product['category'])
product['category'] = _category
_brand = Brand.objects.get(name=product['brand'])
product['brand'] = _brand
new_product = Product(**product)
print(f'Добавлен продукт {new_product}')
new_product.save()
ShopUser.objects.all().delete()
super_user = ShopUser.objects.create_superuser('django',
'*****@*****.**',
'geekbrains',
age=30)
print(f'Пользователь {super_user} создан!\nПароль: geekbrains')
def handle(self, *args, **options):
categories = loadFromJSON('categories')
Category.objects.all().delete()
for category in categories:
new_category = Category(**category)
new_category.save()
books = loadFromJSON('books')
Book.objects.all().delete()
for book in books:
category_name = book['category']
_category = Category.objects.get(name=category_name)
book['category'] = _category
new_book = Book(**book)
new_book.save()
ShopUser.objects.all().delete()
super_user = ShopUser.objects.create_superuser('lulik',
'*****@*****.**',
'luliksuperuser',
age=30)
def handle(self, *args, **kwargs):
data = loadFromJSON('data')
Category.objects.all().delete()
Author.objects.all().delete()
Item.objects.all().delete()
for rec in data:
if rec['model'] == 'mainapp.category':
new_category = Category(**rec['fields'])
new_category.save()
if rec['model'] == 'mainapp.author':
new_author = Author(**rec['fields'])
new_author.save()
for rec in data:
if rec['model'] == 'mainapp.item':
category_id = rec['fields']['category']
author_id = rec['fields']['author']
_category = Category.objects.all()[0]
_author = Author.objects.all()[0]
rec['fields']['category'] = _category
rec['fields']['author'] = _author
new_item = Item(**rec['fields'])
new_item.save()
def get_category(self, category_name):
category = Category.objects.filter(buyer__in=(
DEFAULT_USER,
self.request.user,
)).filter(name=category_name).order_by('-buyer').first()
if not category:
category = Category()
category.buyer = self.request.user
category.name = category_name
last_category = Category.objects.filter(
buyer_id=category.buyer).order_by('-category_id').first()
category.category_id = last_category.category_id + 1 if last_category else 1
category.save()
return category
def handle(self, *args, **options):
clear = input('Clear databases in mainapp (y/n)? ').lower()
if clear == 'y':
CatalogMenu.objects.all().delete()
MainMenu.objects.all().delete()
NewMenu.objects.all().delete()
Product.objects.all().delete()
Address.objects.all().delete()
Category.objects.all().delete()
for i in (
'UPDATE SQLITE_SEQUENCE SET SEQ= 0 WHERE NAME="mainapp_category";',
'UPDATE SQLITE_SEQUENCE SET SEQ= 0 WHERE NAME="mainapp_product";',
'UPDATE SQLITE_SEQUENCE SET SEQ= 0 WHERE NAME="mainapp_mainmenu";',
'UPDATE SQLITE_SEQUENCE SET SEQ= 0 WHERE NAME="mainapp_catalogmenu";',
'UPDATE SQLITE_SEQUENCE SET SEQ= 0 WHERE NAME="mainapp_newmenu";',
'UPDATE SQLITE_SEQUENCE SET SEQ= 0 WHERE NAME="mainapp_address";',
'UPDATE SQLITE_SEQUENCE SET SEQ= 0 WHERE NAME="mainapp_product_category";'
):
connection.cursor().execute(i)
su = input('Create superuser (y/n)? ').lower()
if su == 'y':
CustomUser.objects.create_superuser(
input('Введите имя суперпользователя: '),
'*****@*****.**',
input('Введите пароль суперпользователя: '),
age=18)
categories = load_from_json('all_categories')
for category in categories:
new_category = Category(**category)
new_category.save()
products = load_from_json('all_products')
for product_data in products:
category_names = product_data['category']
product_data.pop('category')
new_product = Product.objects.create(**product_data)
for category_name in category_names:
_category = Category.objects.get(name=category_name)
new_product.category.add(_category)
main_menu = load_from_json('main_menu')
for link in main_menu:
new_link = MainMenu(**link)
new_link.save()
new_menu = load_from_json('new_menu')
for link in new_menu:
category = Category.objects.get(name=link['category'])
link['category'] = category
new_link = NewMenu(**link)
new_link.save()
catalog_menu = load_from_json('catalog_menu')
for link in catalog_menu:
category = Category.objects.get(name=link['category'])
link['category'] = category
new_link = CatalogMenu(**link)
new_link.save()
addresses = load_from_json('all_addresses')
for link in addresses:
new_address = Address(**link)
new_address.save()
def handle(self, *args, **options):
# Подгружаем из файлов суперпользователей (su) и пользователей (buyer)
su_data = settings.CONFIG.get('SUPERUSER', {})
superuser = {
'username': su_data.get('USERNAME', 'admin'),
'email': su_data.get('EMAIL', '*****@*****.**'),
'password': su_data.get('PASSWORD', 'y1u2i3o4y1u2i3o4')
}
buyers = load_from_json('buyers')
Buyer.objects.all().delete()
# Создаем суперпользователя
Buyer.objects.create_superuser(**superuser)
print('SU created')
# Создаем пользователей
for buyer in buyers:
Buyer.objects.create_user(**buyer)
print('Тестовые пользователи подгружены!')
# Подгружаем категории товара из списка
categories = load_from_json('categories')
Category.objects.all().delete()
category_objs = []
for category in categories:
buyer_name = category['buyer']
# Получаем пользователя "создавшего" категорию по имени
_buyer = Buyer.objects.filter(username=buyer_name).first()
# Заменяем имя пользователя идентификатором из базы
category['buyer'] = _buyer
category_objs.append(Category(**category))
Category.objects.bulk_create(category_objs)
print('Каталоги товаров подгружены!')
# Подгружаем список товаров из списка
items = load_from_json('items')
Item.objects.all().delete()
# Так как я не понял зачем нам item_id и с чем он должен связываться, решил генерить его по порядку с нуля
item_id = 0
item_objs = []
for item in items:
buyer_name = item['buyer']
category_name = item['category']
# Получаем пользователя "создавшего" категорию и саму категорию по именам
_buyer = Buyer.objects.filter(username=buyer_name).first()
_category = Category.objects.filter(name=category_name).first()
# Заменяем имя пользователя и название категории идентификаторами из базы
item['buyer'] = _buyer
item['category'] = _category
item['item_id'] = item_id
item_objs.append(Item(**item))
item_id = item_id + 1
Item.objects.bulk_create(item_objs)
print('Товары подгружены!')
print('Заполнение базы завершено!')