Django 脚本调用Model类

此脚步可以放在任意的App下面.

import os
import sys

import django
BASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
print(BASE_DIR)
sys.path.append(BASE_DIR)
os.environ.setdefault("DJANGO_SETTINGS_MODULE", "ERP2.settings")  #ERP2 这个为项目的名
django.setup()

from django.db import transaction
from warehouse.models import  StockPicking


def handel_data():
    picking_queryset = StockPicking.objects.filter(
        id__in=[27787693,27937695])
    for picking_obj in picking_queryset:
        print(picking_obj.id)

    return {'code': 200}




if __name__ == '__main__':
    print('Begin')
    with transaction.atomic():
        save_id = transaction.savepoint()
        try:
            result = handel_data()
        except Exception as e:
            result = {'code': 400, 'msg': 'ERROR:{}'.format(str(e))}
        if result['code'] == 400:
            transaction.savepoint_rollback(save_id)
    print(result)
    print('END')