[aur-dev] [PATCH 1/4] remove mysql dependency from gendummydata
elij
elij.mx at gmail.com
Wed Apr 6 22:23:42 EDT 2011
- remove need to use mysql for escaping the sql -- removing single quote
should be enough
- instead of using sql to fetch categories from a live database, simply
consider categories an integer range, specified to the size of that in the
aur-schema.
---
support/schema/gendummydata.py | 43 +--------------------------------------
1 files changed, 2 insertions(+), 41 deletions(-)
diff --git a/support/schema/gendummydata.py b/support/schema/gendummydata.py
index 7b1d0cf..4dc0de1 100755
--- a/support/schema/gendummydata.py
+++ b/support/schema/gendummydata.py
@@ -56,33 +56,10 @@ if not os.path.exists(SEED_FILE):
sys.stderr.write("Please install the 'words' Arch package\n");
raise SystemExit
-# Make sure database access will be available
-#
-try:
- import MySQLdb
-except:
- sys.stderr.write("Please install the 'mysql-python' Arch package\n");
- raise SystemExit
-
-# try to connect to database
-#
-try:
- db = MySQLdb.connect(host = DB_HOST, user = DB_USER,
- db = DB_NAME, passwd = DB_PASS)
- dbc = db.cursor()
-except:
- sys.stderr.write("Could not connect to database\n");
- raise SystemExit
-
-esc = db.escape_string
-
-
# track what users/package names have been used
#
seen_users = {}
seen_pkgs = {}
-categories = {}
-category_keys = []
user_keys = []
# some functions to generate random data
@@ -95,7 +72,7 @@ def genVersion():
ver.append("%d" % random.randrange(0,100))
return ".".join(ver) + "-u%d" % random.randrange(1,11)
def genCategory():
- return categories[category_keys[random.randrange(0,len(category_keys))]]
+ return random.randrange(0,CATEGORIES_COUNT)
def genUID():
return seen_users[user_keys[random.randrange(0,len(user_keys))]]
@@ -149,22 +126,6 @@ while len(seen_pkgs) < MAX_PKGS:
#
contents = None
-# Load package categories from database
-#
-if DBUG: print "Loading package categories..."
-q = "SELECT * FROM PackageCategories"
-dbc.execute(q)
-row = dbc.fetchone()
-while row:
- categories[row[1]] = row[0]
- row = dbc.fetchone()
-category_keys = categories.keys()
-
-# done with the database
-#
-dbc.close()
-db.close()
-
# developer/tu IDs
#
developers = []
@@ -245,7 +206,7 @@ for p in seen_pkgs.keys():
#
num_comments = random.randrange(PKG_CMNTS[0], PKG_CMNTS[1])
for i in range(0, num_comments):
- fortune = esc(commands.getoutput(FORTUNE_CMD).replace("'",""))
+ fortune = commands.getoutput(FORTUNE_CMD).replace("'","")
now = NOW + random.randrange(400, 86400*3)
s = "INSERT INTO PackageComments (PackageID, UsersID, Comments, CommentTS) VALUES (%d, %d, '%s', %d);\n" % (seen_pkgs[p], genUID(), fortune, now)
out.write(s)
--
1.7.4.1
More information about the aur-dev
mailing list