pythonを触る

Excelも使いたいから、openpyxlを入れておく。

main.py

import sys

def main():
    args = sys.argv
    print(args)

main()

出力

# python main.py aaa bbbb dddd eeeee=aaa
['main.py', 'aaa', 'bbbb', 'dddd', 'eeeee=aaa']

koduka

vscodeのフォーマッターを入れたかったから、autopep8を入れる。

koduka

辞書型のFor文。

main.py

for data in table.datas:
    print(data)
    for key, value in data.items():
        print("key: {}, value: ".format(key, value))

出力

# python main.py /python/assets/test.csv
[{'1': 'aaa', '2': 'bbb', '3': 'ccc', '4': 'ddd', '5': 'eee', '6': 'fff'}, {'1': 'aaa', '2': 'bbb', '3': 'ccc', '4': 'ddd', '5': 'eee', '6': 'fff'}]
{'1': 'aaa', '2': 'bbb', '3': 'ccc', '4': 'ddd', '5': 'eee', '6': 'fff'}
key: 1, value: 
key: 2, value: 
key: 3, value: 
key: 4, value: 
key: 5, value: 
key: 6, value:

koduka

csvの値が[table].fieldの場合、DB接続して、そのテーブルのカラムの値を取得する。
だから、文字列が[table].fieldの場合に、[]で囲んだtableを取り出したい。

まず、[table].fieldのような形式であるかをチェックをして、
OKなら、パース処理を行う。
NGなら、ただの文字列として処理を続ける。

koduka

csvの値が[table].fieldの場合、DB接続して、そのテーブルのカラムの値を取得する。
だから、文字列が[table].fieldの場合に、[]で囲んだtableを取り出したい。

上記、ChatGPTに聞いて、以下を採用！

import re

text = "[table].field"
match = re.match(r'\[([^]]+)\]\.field', text)

if match:
    table_name = match.group(1)
    print(table_name)
else:
    print("マッチしませんでした。")

koduka

パースした結果が、形式通りならDBに接続するようにする。

koduka

環境変数の取得。

koduka

main.py

self.connection = psycopg2.connect("host={} dbname={} user={} password={}".format(
    os.environ.get('DB_HOST', 'localhost'),
    os.environ.get('POSTGRES_USER', 'docker'),
    os.environ.get('POSTGRES_USER', 'docker'),
    os.environ.get('POSTGRES_PASSWORD', 'password'),
))

出力

...
psycopg2.OperationalError: connection to server at "host.docker.internal" (192.168.65.254), port 5432 failed: Connection refused
        Is the server running on that host and accepting TCP/IP connections?

うまくないなー。

koduka

# docker-compose logs postgres
csv-filler-postgres-1  | bash: warning: setlocale: LC_ALL: cannot change locale (ja_JP.UTF-8)
csv-filler-postgres-1  | bash: warning: setlocale: LC_ALL: cannot change locale (ja_JP.UTF-8)
csv-filler-postgres-1  | The files belonging to this database system will be owned by user "postgres".
csv-filler-postgres-1  | This user must also own the server process.
csv-filler-postgres-1  | 
csv-filler-postgres-1  | 
csv-filler-postgres-1  | initdb: error: invalid locale settings; check LANG and LC_* environment variables

Dockefile

FROM postgres

ENV TZ Asia/Tokyo
ENV LANG ja_JP.UTF-8
ENV LANGUAGE ja_JP:ja
ENV LC_ALL ja_JP.UTF-8

環境変数の設定がダメそうなのはわかるけど、解決法がわからない。

koduka

Dockerfile

RUN apt-get update && apt-get install -y locales \
    && rm -rf /var/lib/apt/lists/* \
    && localedef -i ja_JP -c -f UTF-8 -A /usr/share/locale/locale.alias ja_JP.UTF-8

↑を追加する。
Dockerイメージのpostgresベースイメージは、
デフォルトでen_US.utf8のロケールを使用しているため、
initdbを実行する前にロケールデータを追加する必要があるらしい。

koduka

テストしたいから、DBにモックデータを簡単に入れるツールを探す必要があるな。

このスクラップは2024/05/07にクローズされました