summaryrefslogtreecommitdiff
path: root/convert.py
diff options
context:
space:
mode:
Diffstat (limited to 'convert.py')
-rw-r--r--convert.py74
1 files changed, 74 insertions, 0 deletions
diff --git a/convert.py b/convert.py
new file mode 100644
index 0000000..9e8bba4
--- /dev/null
+++ b/convert.py
@@ -0,0 +1,74 @@
+import MySQLdb as mysql
+
+def create():
+ try:
+ con = mysql.connect('localhost', 'wordnet', 'words', 'wordnet');
+
+ with con:
+ cur = con.cursor(mysql.cursors.DictCursor)
+ cur.execute("CREATE TABLE types (id int not null auto_increment, type text, abbreviation text, primary key(id))")
+ cur.execute("INSERT INTO types (type, abbreviation) VALUES('noun', 'n')")
+ cur.execute("INSERT INTO types (type, abbreviation) VALUES('verb', 'v')")
+ cur.execute("INSERT INTO types (type, abbreviation) VALUES('adjective', 'adj')")
+ cur.execute("INSERT INTO types (type, abbreviation) VALUES('adverb', 'adv')")
+ cur.execute("CREATE TABLE definitions (id bigint not null auto_increment, word text, type_id int, sub_id int, synset_id bigint, definition text, primary key(id))")
+ cur.execute("CREATE TABLE uses (id bigint not null auto_increment, definition_id bigint, quote text, primary key(id))")
+
+ except mysql.Error, e:
+ print "Database Error %d: %s" % (e.args[0],e.args[1])
+ sys.exit(1)
+
+def select():
+ try:
+ con = mysql.connect('localhost', 'wordnet', 'words', 'wordnet');
+
+ with con:
+ cur = con.cursor(mysql.cursors.DictCursor)
+ cur.execute("SELECT lemma, pos, sensenum, synsetid, definition, sampleset from dict")
+ rows = cur.fetchall()
+ print len(rows)
+ except mysql.Error, e:
+ print "Database Error %d: %s" % (e.args[0],e.args[1])
+ sys.exit(1)
+
+ return rows
+
+def insert(rows):
+ try:
+ con = mysql.connect('localhost', 'wordnet', 'words', 'wordnet');
+
+ with con:
+ cur = con.cursor(mysql.cursors.DictCursor)
+
+ for row in rows:
+ if row['pos'] == 'n':
+ type_id = 1
+ elif row['pos'] == 'v':
+ type_id = 2
+ elif row['pos'] == 'a' or row['pos'] == 's':
+ type_id = 3
+ elif row['pos'] == 'r':
+ type_id = 4
+
+ cur.execute("INSERT INTO definitions(word, type_id, sub_id, synset_id, definition) values(%s, %s, %s, %s, %s)", [row['lemma'], type_id, row['sensenum'], row['synsetid'], row['definition']])
+
+ row_id = cur.lastrowid
+
+ if row['sampleset'] is not None:
+ uses = row['sampleset'].split("|")
+
+ for use in uses:
+ cur.execute("INSERT INTO uses(definition_id, quote) values(%s, %s)", [row_id, use])
+
+ except mysql.Error, e:
+ print "Database Error %d: %s" % (e.args[0],e.args[1])
+ sys.exit(1)
+
+def main():
+ create()
+ items = select()
+ insert(items)
+
+if __name__ == "__main__":
+ main()
+