populate.py

import os
os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'AF.settings')

import django
django.setup()

from auroreformosa.models import *
import glob
from django.core.files import File
import urllib.parse

def populate():
    user()
    category()
    numero()
    image()

def category():
    add_category('history', 'Histoire', '歷史')
    add_category('culture', 'Culture', '文化')
    add_category('politic', 'Politique', '政治')
    add_category('society', 'Société', '社會')
    add_category('recipe', 'Recette', '食譜')
    add_category('art', 'Art', '藝術')
    add_category('news', 'Nouvelles', '最新消息')
    add_category('edito', 'edito', '編者言')
    add_category('comics', 'Bande dessinée', '漫畫')

def add_category(category_, category_fr, category_tw):
    [c,c_bool] = Category.objects.get_or_create(category = category_)
    c.save()
    [c_tw, c_tw_bool] = CategoryDetail.objects.get_or_create(title=category_tw)
    c_tw.category=c
    c_tw.language='tw'
    c_tw.save()
    [c_fr, c_fr_bool] = CategoryDetail.objects.get_or_create(title=category_fr)
    c_fr.category=c
    c_fr.language='fr'
    c_fr.save()

def user():
    add_user("turtlelin", "林皆安", "chieh-an.lin@cea.fr")

def add_user(username, name, email):
    user, b = User.objects.get_or_create(username = username, email=email)
    user.save()
    userProfile, b = UserProfile.objects.get_or_create(user = user)
    userProfile.name = name
    userProfile.save()

def numero():
    add_numero(1, "Taïwan, un carrefour culturel", "台灣，文化的十字路口")
    add_numero(2, "L’An arrive", "年來了")
    add_numero(3, "Le Riz", "米")
    add_numero(4, "", "")
    add_numero(5, "", "")
    add_numero(6, "", "")
    add_numero(6.1, "", "")

def add_numero(numero_, titleFR, titleTW):
    n, b = Numero.objects.get_or_create(numero = numero_)
    n.titleFR = titleFR
    n.titleTW = titleTW
    n.save()

def image():
    imageList = glob.glob("../../Desktop/img/*")
    for url in imageList:
        imgTitle = url.split("/")[-1]
        img, b = Img.objects.get_or_create(title = imgTitle)
        if b :
            img.imgfile.save(imgTitle, File(open(url, 'rb')), save = True)

if __name__=='__main__':
    populate()
    print('okey')