ComicRocketServer/addComment.py at master · csComicRocket/ComicRocketServer · GitHub

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
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
import sys, random, collections
import textGen
import pymysql
import png
import platform
import time
from switch import switch
## run from cron (probably)

# add comments to existing pages, with probability frequency
# based on how fresh the page is
def AddXKCD(cur):
    pass

def AddDoomsDayMyDear(cur):

    #find site ID
    cur.execute("SELECT id FROM sites WHERE class = 'DoomsDayMyDear'")
    for row in cur.fetchall():
        siteId, = row
        break


    for i in range(30):

        #get last page
        cur.execute("SELECT * FROM pages WHERE site_id = {0} AND next_id IS NULL".format(siteId))
        print(cur.description)
        nextPageExists = False
        for row in cur.fetchall():
            nextPageExists = True
            id, site_id, next_id, uri_path = row
            page = {'id': id, 'site_id': site_id, 'next_id': next_id, 'uri_path': uri_path}
            break

        #loop until coment is added or we reach end of pages
        commentAdded = False
        while not commentAdded and nextPageExists:
            #we stop here at this page, otherwise we delve deeper
            if random.randint(0, 1) == 1:
                username = textGen.BuildWord(random.randint(6,16))
                title = textGen.BuildPhrase(random.randint(1,5), False)
                message = textGen.BuildParagraph(random.randint(1,5))

                #add link to 25%
                if random.randint(0,3) == 0:
                    link = '<a href="http://www.{0}.com">{1}</a>'.format(textGen.BuildWord(random.randint(6,14)), textGen.BuildPhrase(random.randint(1,4)))
                    message += link
                    print(link)

                execution_text = "INSERT INTO `comments` SET `id`=NULL, `page_id`={0}, `datetime`=NOW(), `username`='{1}', `title`='{2}', `body`='{3}', `avatar`=''".format(page['id'], username, title, message)
                cur.execute(execution_text)
                commentAdded = True

            #we get the next oldest page
            else :
                cur.execute("SELECT * FROM pages WHERE site_id = {0} AND next_id = {1}".format(siteId, page['id']))
                nextPageExists = False
                for row in cur.fetchall():
                    nextPageExists = True
                    id, site_id, next_id, uri_path = row
                    page = {'id': id, 'site_id': site_id, 'next_id': next_id, 'uri_path': uri_path}
                    break

        if commentAdded:
            print("Comment added to page_id:{0}".format(page['id']))
        else:
            print("Comment not added")

        time.sleep(2)

    return

if __name__ == '__main__':
    site = sys.argv[1]

    #connect to mysql
    conn = pymysql.connect(host='localhost', port=3306, user='ohmu', passwd='TGSTGSTGS', db='crts')
    cur = conn.cursor()

    for case in switch(site):
        if case('XKCD'):
            AddXKCD(cur)
            break
        if case('DoomsDayMyDear'):
            AddDoomsDayMyDear(cur)
            break
        if case():
            print ("That site isn't recognized!")
            break
        #more cases here

    #close connections
    cur.close()
    conn.close()