kenken999 commited on
Commit
3cc805b
1 Parent(s): 84ccc62
Files changed (1) hide show
  1. polls/databases/insert_line.py +5 -3
polls/databases/insert_line.py CHANGED
@@ -1,6 +1,6 @@
1
  import pandas as pd
2
  import re
3
- from sqlalchemy import create_engine, inspect, Table, MetaData, Column, String, text
4
  import psycopg2
5
 
6
  # チャット履歴ファイルを読み込む関数(最初の3行をスキップ)
@@ -41,7 +41,7 @@ def connect_to_db():
41
  return conn
42
 
43
  # ファイルパスの設定
44
- input_file_path = '/home/user/app/polls/databases/test.csv' # エクスポートされたチャット履歴ファイルのパス
45
 
46
  # チャット履歴を読み込み、特定の列の個人情報をマスク
47
  chat_history_df = load_chat_history(input_file_path)
@@ -55,7 +55,9 @@ engine = create_engine('postgresql+psycopg2://miyataken999:yz1wPf4KrWTm@ep-odd-m
55
  if not inspect(engine).has_table("fasis_chat_history"):
56
  metadata = MetaData()
57
  Table('fasis_chat_history', metadata,
58
- *(Column(name, String) for name in masked_chat_history_df.columns))
 
 
59
  metadata.create_all(engine)
60
 
61
  # マスクされたデータをPostgreSQLにインポート
 
1
  import pandas as pd
2
  import re
3
+ from sqlalchemy import create_engine, inspect, Table, MetaData, Column, Integer, String, text, Sequence
4
  import psycopg2
5
 
6
  # チャット履歴ファイルを読み込む関数(最初の3行をスキップ)
 
41
  return conn
42
 
43
  # ファイルパスの設定
44
+ input_file_path = 'path/to/your/chat_history.csv' # エクスポートされたチャット履歴ファイルのパス
45
 
46
  # チャット履歴を読み込み、特定の列の個人情報をマスク
47
  chat_history_df = load_chat_history(input_file_path)
 
55
  if not inspect(engine).has_table("fasis_chat_history"):
56
  metadata = MetaData()
57
  Table('fasis_chat_history', metadata,
58
+ Column('id', Integer, Sequence('user_id_seq'), primary_key=True, autoincrement=True),
59
+ *(Column(name, String) for name in masked_chat_history_df.columns),
60
+ Column('AnswerField', String))
61
  metadata.create_all(engine)
62
 
63
  # マスクされたデータをPostgreSQLにインポート