Home | History | Annotate | Download | only in Launcher3 
      1 #!/usr/bin/env python2.5
      2 
      3 import cgi
      4 import codecs
      5 import os
      6 import pprint
      7 import re
      8 import shutil
      9 import sys
     10 import sqlite3
     11 
     12 SCREENS = 0
     13 COLUMNS = 4
     14 ROWS = 4
     15 HOTSEAT_SIZE = 4
     16 CELL_SIZE = 110
     17 
     18 CONTAINER_DESKTOP = -100
     19 CONTAINER_HOTSEAT = -101
     20 
     21 DIR = "db_files"
     22 AUTO_FILE = DIR + "/launcher.db"
     23 INDEX_FILE = DIR + "/index.html"
     24 
     25 def usage():
     26   print "usage: print_db.py launcher.db <4x4|5x5|5x6|...> -- prints a launcher.db with"
     27   print "       the specified grid size (rows x cols)"
     28   print "usage: print_db.py <4x4|5x5|5x6|...> -- adb pulls a launcher.db from a device"
     29   print "       and prints it with the specified grid size (rows x cols)"
     30   print
     31   print "The dump will be created in a directory called db_files in cwd."
     32   print "This script will delete any db_files directory you have now"
     33 
     34 
     35 def make_dir():
     36   shutil.rmtree(DIR, True)
     37   os.makedirs(DIR)
     38 
     39 def adb_root_remount():
     40   os.system("adb root")
     41   os.system("adb remount")
     42 
     43 def pull_file(fn):
     44   print "pull_file: " + fn
     45   rv = os.system("adb pull"
     46     + " /data/data/com.android.launcher3/databases/launcher.db"
     47     + " " + fn);
     48   if rv != 0:
     49     print "adb pull failed"
     50     sys.exit(1)
     51 
     52 def get_favorites(conn):
     53   c = conn.cursor()
     54   c.execute("SELECT * FROM favorites")
     55   columns = [d[0] for d in c.description]
     56   rows = []
     57   for row in c:
     58     rows.append(row)
     59   return columns,rows
     60 
     61 def get_screens(conn):
     62   c = conn.cursor()
     63   c.execute("SELECT * FROM workspaceScreens")
     64   columns = [d[0] for d in c.description]
     65   rows = []
     66   for row in c:
     67     rows.append(row)
     68   return columns,rows
     69 
     70 def print_intent(out, id, i, cell):
     71   if cell:
     72     out.write("""<span class="intent" title="%s">shortcut</span>""" % (
     73         cgi.escape(cell, True)
     74       ))
     75 
     76 
     77 def print_icon(out, id, i, cell):
     78   if cell:
     79     icon_fn = "icon_%d.png" % id
     80     out.write("""<img style="width: 3em; height: 3em;" src="%s">""" % ( icon_fn ))
     81     f = file(DIR + "/" + icon_fn, "w")
     82     f.write(cell)
     83     f.close()
     84 
     85 def print_icon_type(out, id, i, cell):
     86   if cell == 0:
     87     out.write("Application (%d)" % cell)
     88   elif cell == 1:
     89     out.write("Shortcut (%d)" % cell)
     90   elif cell == 2:
     91     out.write("Folder (%d)" % cell)
     92   elif cell == 4:
     93     out.write("Widget (%d)" % cell)
     94   elif cell:
     95     out.write("%d" % cell)
     96 
     97 def print_cell(out, id, i, cell):
     98   if not cell is None:
     99     out.write(cgi.escape(unicode(cell)))
    100 
    101 FUNCTIONS = {
    102   "intent": print_intent,
    103   "icon": print_icon,
    104   "iconType": print_icon_type
    105 }
    106 
    107 def render_cell_info(out, cell, occupied):
    108   if cell is None:
    109     out.write("    <td width=%d height=%d></td>\n" %
    110         (CELL_SIZE, CELL_SIZE))
    111   elif cell == occupied:
    112     pass
    113   else:
    114     cellX = cell["cellX"]
    115     cellY = cell["cellY"]
    116     spanX = cell["spanX"]
    117     spanY = cell["spanY"]
    118     intent = cell["intent"]
    119     if intent:
    120       title = "title=\"%s\"" % cgi.escape(cell["intent"], True)
    121     else:
    122       title = ""
    123     out.write(("    <td colspan=%d rowspan=%d width=%d height=%d"
    124         + " bgcolor=#dddddd align=center valign=middle %s>") % (
    125           spanX, spanY,
    126           (CELL_SIZE*spanX), (CELL_SIZE*spanY),
    127           title))
    128     itemType = cell["itemType"]
    129     if itemType == 0:
    130       out.write("""<img style="width: 4em; height: 4em;" src="icon_%d.png">\n""" % ( cell["_id"] ))
    131       out.write("<br/>\n")
    132       out.write(cgi.escape(cell["title"]) + " <br/><i>(app)</i>")
    133     elif itemType == 1:
    134       out.write("""<img style="width: 4em; height: 4em;" src="icon_%d.png">\n""" % ( cell["_id"] ))
    135       out.write("<br/>\n")
    136       out.write(cgi.escape(cell["title"]) + " <br/><i>(shortcut)</i>")
    137     elif itemType == 2:
    138       out.write("""<i>folder</i>""")
    139     elif itemType == 4:
    140       out.write("<i>widget %d</i><br/>\n" % cell["appWidgetId"])
    141     else:
    142       out.write("<b>unknown type: %d</b>" % itemType)
    143     out.write("</td>\n")
    144 
    145 def render_screen_info(out, screen):
    146   out.write("<tr>")
    147   out.write("<td>%s</td>" % (screen["_id"]))
    148   out.write("<td>%s</td>" % (screen["screenRank"]))
    149   out.write("</tr>")
    150 
    151 def process_file(fn):
    152   global SCREENS, COLUMNS, ROWS, HOTSEAT_SIZE
    153   print "process_file: " + fn
    154   conn = sqlite3.connect(fn)
    155   columns,rows = get_favorites(conn)
    156   screenCols, screenRows = get_screens(conn)
    157 
    158   data = [dict(zip(columns,row)) for row in rows]
    159   screenData = [dict(zip(screenCols, screenRow)) for screenRow in screenRows]
    160 
    161   # Calculate the proper number of screens, columns, and rows in this db
    162   screensIdMap = []
    163   hotseatIdMap = []
    164   HOTSEAT_SIZE = 0
    165   for d in data:
    166     if d["spanX"] is None:
    167       d["spanX"] = 1
    168     if d["spanY"] is None:
    169       d["spanY"] = 1
    170     if d["container"] == CONTAINER_DESKTOP:
    171       if d["screen"] not in screensIdMap:
    172         screensIdMap.append(d["screen"])
    173       COLUMNS = max(COLUMNS, d["cellX"] + d["spanX"])
    174       ROWS = max(ROWS, d["cellX"] + d["spanX"])
    175     elif d["container"] == CONTAINER_HOTSEAT:
    176       hotseatIdMap.append(d["screen"])
    177       HOTSEAT_SIZE = max(HOTSEAT_SIZE, d["screen"] + 1)
    178   SCREENS = len(screensIdMap)
    179 
    180   out = codecs.open(INDEX_FILE, encoding="utf-8", mode="w")
    181   out.write("""<html>
    182 <head>
    183 <style type="text/css">
    184 .intent {
    185   font-style: italic;
    186 }
    187 </style>
    188 </head>
    189 <body>
    190 """)
    191 
    192   # Data table
    193   out.write("<b>Favorites table</b><br/>\n")
    194   out.write("""<html>
    195 <table border=1 cellspacing=0 cellpadding=4>
    196 <tr>
    197 """)
    198   print_functions = []
    199   for col in columns:
    200     print_functions.append(FUNCTIONS.get(col, print_cell))
    201   for i in range(0,len(columns)):
    202     col = columns[i]
    203     out.write("""  <th>%s</th>
    204 """ % ( col ))
    205   out.write("""
    206 </tr>
    207 """)
    208 
    209   for row in rows:
    210     out.write("""<tr>
    211 """)
    212     for i in range(0,len(row)):
    213       cell = row[i]
    214       # row[0] is always _id
    215       out.write("""  <td>""")
    216       print_functions[i](out, row[0], row, cell)
    217       out.write("""</td>
    218 """)
    219     out.write("""</tr>
    220 """)
    221   out.write("""</table>
    222 """)
    223 
    224   # Screens
    225   out.write("<br/><b>Screens</b><br/>\n")
    226   out.write("<table class=layout border=1 cellspacing=0 cellpadding=4>\n")
    227   out.write("<tr><td>Screen ID</td><td>Rank</td></tr>\n")
    228   for screen in screenData:
    229     render_screen_info(out, screen)
    230   out.write("</table>\n")
    231 
    232   # Hotseat
    233   hotseat = []
    234   for i in range(0, HOTSEAT_SIZE):
    235     hotseat.append(None)
    236   for row in data:
    237     if row["container"] != CONTAINER_HOTSEAT:
    238       continue
    239     screen = row["screen"]
    240     hotseat[screen] = row
    241   out.write("<br/><b>Hotseat</b><br/>\n")
    242   out.write("<table class=layout border=1 cellspacing=0 cellpadding=4>\n")
    243   for cell in hotseat:
    244     render_cell_info(out, cell, None)
    245   out.write("</table>\n")
    246 
    247   # Pages
    248   screens = []
    249   for i in range(0,SCREENS):
    250     screen = []
    251     for j in range(0,ROWS):
    252       m = []
    253       for k in range(0,COLUMNS):
    254         m.append(None)
    255       screen.append(m)
    256     screens.append(screen)
    257   occupied = "occupied"
    258   for row in data:
    259     # desktop
    260     if row["container"] != CONTAINER_DESKTOP:
    261       continue
    262     screen = screens[screensIdMap.index(row["screen"])]
    263     cellX = row["cellX"]
    264     cellY = row["cellY"]
    265     spanX = row["spanX"]
    266     spanY = row["spanY"]
    267     for j in range(cellY, cellY+spanY):
    268       for k in range(cellX, cellX+spanX):
    269         screen[j][k] = occupied
    270     screen[cellY][cellX] = row
    271   i=0
    272   for screen in screens:
    273     out.write("<br/><b>Screen %d</b><br/>\n" % i)
    274     out.write("<table class=layout border=1 cellspacing=0 cellpadding=4>\n")
    275     for m in screen:
    276       out.write("  <tr>\n")
    277       for cell in m:
    278         render_cell_info(out, cell, occupied)
    279       out.write("</tr>\n")
    280     out.write("</table>\n")
    281     i=i+1
    282 
    283   out.write("""
    284 </body>
    285 </html>
    286 """)
    287 
    288   out.close()
    289 
    290 def updateDeviceClassConstants(str):
    291   global SCREENS, COLUMNS, ROWS, HOTSEAT_SIZE
    292   match = re.search(r"(\d+)x(\d+)", str)
    293   if match:
    294     COLUMNS = int(match.group(1))
    295     ROWS = int(match.group(2))
    296     HOTSEAT_SIZE = 2 * int(COLUMNS / 2)
    297     return True
    298   return False
    299 
    300 def main(argv):
    301   if len(argv) == 1 or (len(argv) == 2 and updateDeviceClassConstants(argv[1])):
    302     make_dir()
    303     adb_root_remount()
    304     pull_file(AUTO_FILE)
    305     process_file(AUTO_FILE)
    306   elif len(argv) == 2 or (len(argv) == 3 and updateDeviceClassConstants(argv[2])):
    307     make_dir()
    308     process_file(argv[1])
    309   else:
    310     usage()
    311 
    312 if __name__=="__main__":
    313   main(sys.argv)
    314