120 lines
5.2 KiB
Python
120 lines
5.2 KiB
Python
from django.core.management.base import BaseCommand
|
|
from apidirectory.models import ApiCategory, ApiItem
|
|
|
|
|
|
class Command(BaseCommand):
|
|
help = 'Seed API directory with initial data'
|
|
|
|
def handle(self, *args, **options):
|
|
self.stdout.write('Seeding API directory...')
|
|
|
|
categories_data = [
|
|
{'name': '数据查询', 'icon': 'SearchOutlined', 'sort_order': 1},
|
|
{'name': '位置服务', 'icon': 'EnvironmentOutlined', 'sort_order': 2},
|
|
{'name': '金融货币', 'icon': 'DollarOutlined', 'sort_order': 3},
|
|
{'name': '工具服务', 'icon': 'ToolOutlined', 'sort_order': 4},
|
|
{'name': '翻译服务', 'icon': 'TranslationOutlined', 'sort_order': 5},
|
|
]
|
|
|
|
categories = {}
|
|
for cat_data in categories_data:
|
|
cat, created = ApiCategory.objects.get_or_create(
|
|
name=cat_data['name'],
|
|
defaults=cat_data
|
|
)
|
|
categories[cat_data['name']] = cat
|
|
if created:
|
|
self.stdout.write(f' Created category: {cat.name}')
|
|
|
|
items_data = [
|
|
{
|
|
'name': '天气查询',
|
|
'description': '根据城市名称查询实时天气信息，支持全球城市和多种温度单位',
|
|
'url_path': '/api/weather',
|
|
'method': 'GET',
|
|
'category': '数据查询',
|
|
'is_new': True,
|
|
'color': '#10b981',
|
|
'image_url': 'https://images.unsplash.com/photo-1504608524841-42fe6f032b4b?w=800&h=600&fit=crop',
|
|
},
|
|
{
|
|
'name': '空气质量指数',
|
|
'description': '查询指定城市的空气质量指数AQI、PM2.5、PM10等环境数据',
|
|
'url_path': '/api/air-quality/',
|
|
'method': 'GET',
|
|
'category': '数据查询',
|
|
'is_new': True,
|
|
'color': '#3b82f6',
|
|
'image_url': 'https://images.unsplash.com/photo-1441974231531-c6227db76b6e?w=800&h=600&fit=crop',
|
|
},
|
|
{
|
|
'name': 'IP地址定位',
|
|
'description': '根据IP地址获取地理位置信息，支持自动识别客户端IP',
|
|
'url_path': '/api/getIpData/',
|
|
'method': 'GET',
|
|
'category': '位置服务',
|
|
'color': '#8b5cf6',
|
|
'image_url': 'https://images.unsplash.com/photo-1451187580459-43490279c0fa?w=800&h=600&fit=crop',
|
|
},
|
|
{
|
|
'name': '二维码生成',
|
|
'description': '根据URL或文本生成二维码图片',
|
|
'url_path': '/api/qrcode-generator/',
|
|
'method': 'POST',
|
|
'category': '工具服务',
|
|
'color': '#ec4899',
|
|
'image_url': 'https://images.unsplash.com/photo-1558591710-4b4a1ae0f04d?w=800&h=600&fit=crop',
|
|
},
|
|
{
|
|
'name': '百度翻译',
|
|
'description': '支持200+语言互译，自动检测源语言',
|
|
'url_path': '/api/baiduFanyi/',
|
|
'method': 'POST',
|
|
'category': '翻译服务',
|
|
'color': '#06b6d4',
|
|
'is_enabled': True,
|
|
'image_url': 'https://images.unsplash.com/photo-1456513080510-7bf3a84b82f8?w=800&h=600&fit=crop',
|
|
},
|
|
{
|
|
'name': '文本比较',
|
|
'description': '比较两段文本的差异，支持多种比较模式',
|
|
'url_path': '/api/tool/text-diff/',
|
|
'method': 'POST',
|
|
'category': '工具服务',
|
|
'color': '#14b8a6',
|
|
'image_url': 'https://images.unsplash.com/photo-1518770660439-4636190af475?w=800&h=600&fit=crop',
|
|
},
|
|
{
|
|
'name': '图片压缩',
|
|
'description': '在线压缩图片，支持批量处理和质量调节',
|
|
'url_path': '/api/tool/image-compressor',
|
|
'method': 'POST',
|
|
'category': '工具服务',
|
|
'color': '#84cc16',
|
|
'image_url': 'https://images.unsplash.com/photo-1542038784456-1ea8e935640e?w=800&h=600&fit=crop',
|
|
},
|
|
{
|
|
'name': '短链接生成',
|
|
'description': '将长链接转换为短链接，支持自定义后缀',
|
|
'url_path': '/api/shorturl/shorten/',
|
|
'method': 'POST',
|
|
'category': '工具服务',
|
|
'color': '#f97316',
|
|
'image_url': 'https://images.unsplash.com/photo-1557682172-298e090bd0f1?w=800&h=600&fit=crop',
|
|
},
|
|
]
|
|
|
|
for item_data in items_data:
|
|
cat_name = item_data.pop('category')
|
|
item_data['category'] = categories[cat_name]
|
|
item, created = ApiItem.objects.update_or_create(
|
|
url_path=item_data['url_path'],
|
|
defaults=item_data
|
|
)
|
|
if created:
|
|
self.stdout.write(f' Created item: {item.name} ({item.url_path})')
|
|
else:
|
|
self.stdout.write(f' Updated item: {item.name} ({item.url_path})')
|
|
|
|
self.stdout.write(self.style.SUCCESS('API directory seeded successfully!'))
|