-
Notifications
You must be signed in to change notification settings - Fork 3
/
Copy pathimport-prompts
executable file
·42 lines (36 loc) · 1.3 KB
/
import-prompts
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
#!/usr/bin/env php
<?php
/* Copyright 2024 Romain "Artefact2" Dal Maso <romain.dalmaso@artefact2.com>
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
require __DIR__.'/common.php';
$json = json_decode(file_get_contents('php://stdin'), true);
$db = get_db();
if($db->exec('BEGIN IMMEDIATE;') === false) {
fprintf(STDERR, "%s: could not begin immediate transaction\n", $argv[1]);
die(1);
}
$sel_stmt = $db->prepare('SELECT prompt_id FROM prompts WHERE prompt = :prompt');
$sel_stmt->bindParam(':prompt', $prompt);
$ins_stmt = $db->prepare('INSERT INTO prompts(prompt) VALUES(:prompt);');
$ins_stmt->bindParam(':prompt', $prompt);
foreach($json as $prompt) {
$prompt = $prompt;
if($sel_stmt->execute()->fetchArray() === false) {
$ins_stmt->execute();
echo '.';
}
}
$db->exec('COMMIT;');
echo PHP_EOL;