diff options
-rw-r--r-- | src/ryuslash/org/dvdroid/DVDDataSource.java | 32 | ||||
-rw-r--r-- | src/ryuslash/org/dvdroid/DVDroidActivity.java | 9 |
2 files changed, 35 insertions, 6 deletions
diff --git a/src/ryuslash/org/dvdroid/DVDDataSource.java b/src/ryuslash/org/dvdroid/DVDDataSource.java index 6658c8d..b9fe166 100644 --- a/src/ryuslash/org/dvdroid/DVDDataSource.java +++ b/src/ryuslash/org/dvdroid/DVDDataSource.java @@ -65,24 +65,48 @@ public class DVDDataSource SQLiteHelper.COLUMN_ID + " = " + id, null); } - public List<DVD> getWishlist() + public List<DVD> getListFromCursor(Cursor cursor) { List<DVD> dvds = new ArrayList<DVD>(); - Cursor cursor = database.query(SQLiteHelper.TABLE_DVD, - allColumns, "wishlist = 1", null, - null, null, "title"); cursor.moveToFirst(); + while(!cursor.isAfterLast()) { DVD dvd = cursorToDVD(cursor); + dvds.add(dvd); cursor.moveToNext(); } cursor.close(); + return dvds; } + public List<DVD> getWishList() + { + Cursor cursor = database.query(SQLiteHelper.TABLE_DVD, + allColumns, "wishlist = 1", null, + null, null, "title"); + return getListFromCursor(cursor); + } + + public List<DVD> getWatchList() + { + Cursor cursor = database.query(SQLiteHelper.TABLE_DVD, + allColumns, "watched = 0", null, + null, null, "title"); + return getListFromCursor(cursor); + } + + public List<DVD> getCollection() + { + Cursor cursor = database.query(SQLiteHelper.TABLE_DVD, + allColumns, "wishlist = 0", null, + null, null, "title"); + return getListFromCursor(cursor); + } + private DVD cursorToDVD(Cursor cursor) { DVD dvd = new DVD(); diff --git a/src/ryuslash/org/dvdroid/DVDroidActivity.java b/src/ryuslash/org/dvdroid/DVDroidActivity.java index e510a7b..533bc62 100644 --- a/src/ryuslash/org/dvdroid/DVDroidActivity.java +++ b/src/ryuslash/org/dvdroid/DVDroidActivity.java @@ -18,6 +18,11 @@ public class DVDroidActivity extends ListActivity { private DVDDataSource datasource; + private List<DVD> getData() + { + return datasource.getCollection(); + } + @Override protected void onActivityResult(int requestCode, int resultCode, @@ -27,7 +32,7 @@ public class DVDroidActivity extends ListActivity datasource.open(); ArrayAdapter<DVD> adapter = (ArrayAdapter<DVD>)getListAdapter(); - List<DVD> dvds = datasource.getWishlist(); + List<DVD> dvds = getData(); adapter.clear(); adapter.addAll(dvds); @@ -44,7 +49,7 @@ public class DVDroidActivity extends ListActivity datasource = new DVDDataSource(this); datasource.open(); - List<DVD> values = datasource.getWishlist(); + List<DVD> values = getData(); ArrayAdapter<DVD> adapter = new ArrayAdapter<DVD>(this, |