0

私はこのような小さなpythonスクリプトを書いています：

#!/usr/bin/env python

from sqlite3 import dbapi2 as sqlite
from sys import argv,exit

db_name = "hashez.db"

def define_db():
    try:
        conn = sqlite.connect(db_name)
    except IOError as e:
        print "problem while creating/connecting the db:",e.args[0]
        exit(1)

    return conn

def write_db(conn,cursor,na,ha):
    conn.execute("CREATE TABLE IF NOT EXISTS user (name TEXT UNIQUE, hash TEXT UNIQUE)")
    query = "INSERT OR REPLACE INTO user VALUES($name,$hash)"
    cursor.execute(query,[na],[ha])
    cursor.close()  
    conn.commit()
    conn.close()
    exit(0)

if __name__ == "__main__":
    if len(argv) == 2:
        na,ha = argv[1]
        #ha = argv[2]
    else:
        print "no argument given - stopping now"
        exit(1)

    conn = define_db()
    cursor = conn.cursor()
    write_db(conn,cursor,na,ha)

1つの入力を取り込もうとすると問題ありません

python user.py blah

しかし、複数でやろうとすると、elseループに入ります。

どこでエラーをしていますか? 案内してください...

4

2 に答える 2

1

最初の引数は実際にはファイル名であるため、このエラーが発生します。

簡単なテストファイルは、これがどのように機能するかを示しています。

[~]$ cat test.py
from sys import argv

if __name__ == '__main__':
    print argv
    print len(argv)
[~]$ python test.py one two
['test.py', 'one', 'two']
3
[~]$ python test.py one
['test.py', 'one']
2

SQL/クエリも修正する必要があります

query = "INSERT OR REPLACE INTO user VALUES(?,?)"
cursor.execute(query,[na,ha])
conn.commit()

sqliteapidocsを参照してください。

于 2012-08-02T22:09:51.393 に答える
0

の最初の要素sys.argvはスクリプトの名前であるため、実際にはifステートメントに次の要素が必要になる可能性があります。

if len(argv) == 3:
    na = argv[1]
    ha = argv[2]
于 2012-08-02T22:09:02.610 に答える