flask sqlalchemy で外部キーで繋がれてる二つのテーブルの情報を取得したい

Question

python flask初学者です。
flask sqlalchemyでのデータの取得について
実現したいことは
外部キーで繋がっている二つのテーブルから
特定のデータを取得してjsonで返したいです。
テーブルはCompany, Employee

class Company(db.Model):

    def __init__(self, c_name, created, modified):
        self.name = c_name
        self.created = created

        employee = relationship('Employee', backref='company', uselist=False)

    __tablename__ = 'company'

    id = db.Column(db.Integer(), primary_key=True, nullable=False, autoincrement=True, unique=True, comment="ID")
    c_name = db.Column(db.String(255), nullable=True)
    created = db.Column(db.Date, nullable=True, default=datetime.utcnow)

class Employee(db.Model):

    def __init__(self, e_name, birthday, gender, company_id):
        self.name = e_name
        self.birthday = birthday
        self.gender = gender
        self.company_id = company_id

    __tablename__ = 'employee'

    id = db.Column(db.Integer(), nullable=False, primary_key=True, autoincrement=True, unique=True, comment="ID")
    e_name = db.Column(db.String(255), nullable=False)
    birthday = db.Column(db.Date, nullable=False)
    gender = db.Column(db.Integer(), nullable=False, comment="性別(0:男、1:女)")
    company_id = db.Column(db.Integer, db.ForeignKey('company.id'), nullable=False)

この二つのテーブルから
例えば
Companyテーブルに会社が5つ、
Employeeテーブルに社員が50人のデータがあります。

やりたいのはCompany id 1に所属している Employeeの男女の人数を取得したいです。
それをjson形式で出力したい。

Company_id = 1 の社員が男6人、女4人の場合
Company_id = 2 の社員が男3人、女7人の場合

{
    "e_gender": [
        {
            "id": 1,
            "name": "c_no1",
            "male": 6,
            "female": 4
        },
        {
            "id": 2,
            "name": "c_no2",
            "male": 3,
            "female": 7
        },
    ]
}

となるように取得したいです。
genderの総数の取得はできたのですが
会社のidごとに分けて取得することができません。

まだ学習を始めて間もないので
ロジックもあまりわかっていないので質問が分かりにくいかと
思いますが、
わかる方、ご教授頂けませんか？
よろしくお願いいたします。

insert追加しました。

engine = create_engine(''mysql+pymysql://{user}:{password}@{host}/flask_sample?charset=utf8'', echo=True)
Session = sessionmaker(bind=engine)
session = Session()
 
e_employee = Employee(name='e_name1', birthday='1990-01-01', 
          gender='0', company_id = company_id)
session.add(e_user)
 
session.commit()

Comments

[FiroProchainezo]

Create文と、中に入れるデータをいただけませんか？

[yuto36]

import os

app = Flask(__name__)
app.config['SQLALCHEMY_DATABASE_URI'] = 'mysql+pymysql://{user}:{password}@{host}/flask_sample?charset=utf8'
app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False

ターミナルから
$ python
>>>from flask_sample import db
>>>db.create_all()
とたたいて、dbのCreateをしています。

中に入れるデータというのは、Company、Employeeのテーブルの中身という解釈で間違ってないでしょうか？
サンプルで作成しているので
テーブルの中身は
Companyテーブルは
id = 1
c_name = c_no1

Employeeテーブルは
id = 1
e_name = e_name1
birthday = 1990-01-01
gender = 0 ('男 ')
company_id = 1
という感じで作成しています。

[FiroProchainezo]

こちらに返信で書くとコードブロックが使えないので、質問を編集していただけませんか？

データはinsert文がほしいです。

[yuto36]

認識不足ですいません。
質問の編集をしました。よろしくお願いします。

[FiroProchainezo]

あれ？これってsqlalchemyですか？
flask使ってますか？

[yuto36]

flask sqlalchemyを使ってます。

Answer 1

きれいじゃありませんが、これで一応期待値通りです。

from flask import Flask
from flask_sqlalchemy import SQLAlchemy
from datetime import datetime

app = Flask(__name__)

db = SQLAlchemy(app)


class Company(db.Model):
    __tablename__ = 'company'

    id = db.Column(db.Integer(), primary_key=True, nullable=False, autoincrement=True, unique=True, comment="ID")
    c_name = db.Column(db.String(255), nullable=True)
    created = db.Column(db.Date, nullable=True, default=datetime.utcnow)
    employee = db.relationship('Employee', backref='company', uselist=False)


class Employee(db.Model):
    __tablename__ = 'employee'
    id = db.Column(db.Integer(), nullable=False, primary_key=True, autoincrement=True, unique=True, comment="ID")
    e_name = db.Column(db.String(255), nullable=False)
    birthday = db.Column(db.Date, nullable=False)
    gender = db.Column(db.Integer(), nullable=False, comment="性別(0:男、1:女)")
    company_id = db.Column(db.Integer, db.ForeignKey('company.id'), nullable=False)


app.config[
    'SQLALCHEMY_DATABASE_URI'] = 'mysql+pymysql://{user}:{password}@{host}:{portnumber}/{database}?charset=utf8'.format(
    user='user',
    password='pass',
    host='localhost',
    portnumber=3306,
    database='database')


@app.route('/')
def hello_world():
    company = db.session.query(Company).filter().all()
    dic = dict()
    dic['e_gender'] = list()
    for n in company:
        tmp = dict()
        tmp['id'] = n.id
        tmp['name'] = n.c_name
        employee = db.session.query(Employee).filter(
            Employee.company_id == n.id,
            Employee.gender == 1
        ).count()
        tmp['male'] = employee
        employee = db.session.query(Employee).filter(
            Employee.company_id == n.id,
            Employee.gender == 0
        ).count()
        tmp['famale'] = employee
        dic['e_gender'].append(tmp)

    return str(dic)


if __name__ == '__main__':
    app.run()