mirror of https://github.com/iSoron/uhabits.git
parent
a0582b65d5
commit
dfe5c4954e
@ -1,213 +0,0 @@
|
|||||||
/*
|
|
||||||
* Copyright (C) 2016 Álinson Santos Xavier <isoron@gmail.com>
|
|
||||||
*
|
|
||||||
* This file is part of Loop Habit Tracker.
|
|
||||||
*
|
|
||||||
* Loop Habit Tracker is free software: you can redistribute it and/or modify
|
|
||||||
* it under the terms of the GNU General Public License as published by the
|
|
||||||
* Free Software Foundation, either version 3 of the License, or (at your
|
|
||||||
* option) any later version.
|
|
||||||
*
|
|
||||||
* Loop Habit Tracker is distributed in the hope that it will be useful, but
|
|
||||||
* WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
|
|
||||||
* or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
|
|
||||||
* more details.
|
|
||||||
*
|
|
||||||
* You should have received a copy of the GNU General Public License along
|
|
||||||
* with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
||||||
*/
|
|
||||||
|
|
||||||
package org.isoron.uhabits.io;
|
|
||||||
|
|
||||||
import android.content.Context;
|
|
||||||
import android.database.Cursor;
|
|
||||||
import android.database.sqlite.SQLiteDatabase;
|
|
||||||
import android.util.Log;
|
|
||||||
|
|
||||||
import com.activeandroid.Cache;
|
|
||||||
|
|
||||||
import org.isoron.helpers.DateHelper;
|
|
||||||
import org.isoron.uhabits.models.Habit;
|
|
||||||
import org.isoron.uhabits.models.Score;
|
|
||||||
|
|
||||||
import java.io.File;
|
|
||||||
import java.io.FileInputStream;
|
|
||||||
import java.io.FileOutputStream;
|
|
||||||
import java.io.FileWriter;
|
|
||||||
import java.io.IOException;
|
|
||||||
import java.text.SimpleDateFormat;
|
|
||||||
import java.util.Date;
|
|
||||||
import java.util.LinkedList;
|
|
||||||
import java.util.List;
|
|
||||||
import java.util.Locale;
|
|
||||||
import java.util.TimeZone;
|
|
||||||
import java.util.zip.ZipEntry;
|
|
||||||
import java.util.zip.ZipOutputStream;
|
|
||||||
|
|
||||||
public class CSVExporter
|
|
||||||
{
|
|
||||||
private List<Habit> habits;
|
|
||||||
private Context context;
|
|
||||||
private java.text.DateFormat dateFormat;
|
|
||||||
|
|
||||||
private List<String> generateDirs;
|
|
||||||
private List<String> generateFilenames;
|
|
||||||
|
|
||||||
private String basePath;
|
|
||||||
|
|
||||||
public CSVExporter(Context context, List<Habit> habits)
|
|
||||||
{
|
|
||||||
this.habits = habits;
|
|
||||||
this.context = context;
|
|
||||||
generateDirs = new LinkedList<>();
|
|
||||||
generateFilenames = new LinkedList<>();
|
|
||||||
|
|
||||||
basePath = String.format("%s/export/", context.getFilesDir());
|
|
||||||
|
|
||||||
dateFormat = new SimpleDateFormat("yyyy-MM-dd", Locale.US);
|
|
||||||
dateFormat.setTimeZone(TimeZone.getTimeZone("UTC"));
|
|
||||||
}
|
|
||||||
|
|
||||||
public String formatDate(long timestamp)
|
|
||||||
{
|
|
||||||
return dateFormat.format(new Date(timestamp));
|
|
||||||
}
|
|
||||||
|
|
||||||
public String formatScore(int score)
|
|
||||||
{
|
|
||||||
return String.format("%.2f", ((float) score) / Score.MAX_VALUE);
|
|
||||||
}
|
|
||||||
|
|
||||||
private void writeScores(String dirPath, Habit habit) throws IOException
|
|
||||||
{
|
|
||||||
String path = dirPath + "scores.csv";
|
|
||||||
FileWriter out = new FileWriter(basePath + path);
|
|
||||||
generateFilenames.add(path);
|
|
||||||
|
|
||||||
String query = "select timestamp, score from score where habit = ? order by timestamp";
|
|
||||||
String params[] = { habit.getId().toString() };
|
|
||||||
|
|
||||||
SQLiteDatabase db = Cache.openDatabase();
|
|
||||||
Cursor cursor = db.rawQuery(query, params);
|
|
||||||
|
|
||||||
if(!cursor.moveToFirst()) return;
|
|
||||||
|
|
||||||
do
|
|
||||||
{
|
|
||||||
String timestamp = formatDate(cursor.getLong(0));
|
|
||||||
String score = formatScore(cursor.getInt(1));
|
|
||||||
out.write(String.format("%s,%s\n", timestamp, score));
|
|
||||||
|
|
||||||
} while(cursor.moveToNext());
|
|
||||||
|
|
||||||
out.close();
|
|
||||||
cursor.close();
|
|
||||||
}
|
|
||||||
|
|
||||||
private void writeCheckmarks(String dirPath, Habit habit) throws IOException
|
|
||||||
{
|
|
||||||
String path = dirPath + "checkmarks.csv";
|
|
||||||
FileWriter out = new FileWriter(basePath + path);
|
|
||||||
generateFilenames.add(path);
|
|
||||||
|
|
||||||
String query = "select timestamp, value from checkmarks where habit = ? order by timestamp";
|
|
||||||
String params[] = { habit.getId().toString() };
|
|
||||||
|
|
||||||
SQLiteDatabase db = Cache.openDatabase();
|
|
||||||
Cursor cursor = db.rawQuery(query, params);
|
|
||||||
|
|
||||||
if(!cursor.moveToFirst()) return;
|
|
||||||
|
|
||||||
do
|
|
||||||
{
|
|
||||||
String timestamp = formatDate(cursor.getLong(0));
|
|
||||||
Integer value = cursor.getInt(1);
|
|
||||||
out.write(String.format("%s,%d\n", timestamp, value));
|
|
||||||
|
|
||||||
} while(cursor.moveToNext());
|
|
||||||
|
|
||||||
out.close();
|
|
||||||
cursor.close();
|
|
||||||
}
|
|
||||||
|
|
||||||
private void writeFiles(Habit habit) throws IOException
|
|
||||||
{
|
|
||||||
String path = String.format("%s/", habit.name);
|
|
||||||
new File(basePath + path).mkdirs();
|
|
||||||
generateDirs.add(path);
|
|
||||||
|
|
||||||
writeScores(path, habit);
|
|
||||||
writeCheckmarks(path, habit);
|
|
||||||
}
|
|
||||||
|
|
||||||
private void writeZipFile(String zipFilename) throws IOException
|
|
||||||
{
|
|
||||||
FileOutputStream fos = new FileOutputStream(zipFilename);
|
|
||||||
ZipOutputStream zos = new ZipOutputStream(fos);
|
|
||||||
|
|
||||||
for(String filename : generateFilenames)
|
|
||||||
addFileToZip(zos, filename);
|
|
||||||
|
|
||||||
zos.close();
|
|
||||||
fos.close();
|
|
||||||
}
|
|
||||||
|
|
||||||
private void addFileToZip(ZipOutputStream zos, String filename) throws IOException
|
|
||||||
{
|
|
||||||
FileInputStream fis = new FileInputStream(new File(basePath + filename));
|
|
||||||
ZipEntry ze = new ZipEntry(filename);
|
|
||||||
zos.putNextEntry(ze);
|
|
||||||
|
|
||||||
int length;
|
|
||||||
byte bytes[] = new byte[1024];
|
|
||||||
while((length = fis.read(bytes)) >= 0)
|
|
||||||
zos.write(bytes, 0, length);
|
|
||||||
|
|
||||||
zos.closeEntry();
|
|
||||||
fis.close();
|
|
||||||
}
|
|
||||||
|
|
||||||
private void cleanup()
|
|
||||||
{
|
|
||||||
for(String filename : generateFilenames)
|
|
||||||
new File(basePath + filename).delete();
|
|
||||||
|
|
||||||
for(String filename : generateDirs)
|
|
||||||
new File(basePath + filename).delete();
|
|
||||||
|
|
||||||
new File(basePath).delete();
|
|
||||||
}
|
|
||||||
|
|
||||||
public String writeArchive()
|
|
||||||
{
|
|
||||||
String date = formatDate(DateHelper.getStartOfToday());
|
|
||||||
|
|
||||||
File dir = context.getExternalCacheDir();
|
|
||||||
|
|
||||||
if(dir == null)
|
|
||||||
{
|
|
||||||
Log.e("CSVExporter", "No suitable directory found.");
|
|
||||||
return null;
|
|
||||||
}
|
|
||||||
|
|
||||||
String zipFilename = String.format("%s/habits-%s.zip", dir, date);
|
|
||||||
|
|
||||||
try
|
|
||||||
{
|
|
||||||
for (Habit h : habits)
|
|
||||||
writeFiles(h);
|
|
||||||
|
|
||||||
writeZipFile(zipFilename);
|
|
||||||
cleanup();
|
|
||||||
}
|
|
||||||
catch (IOException e)
|
|
||||||
{
|
|
||||||
e.printStackTrace();
|
|
||||||
return null;
|
|
||||||
}
|
|
||||||
|
|
||||||
return zipFilename;
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
}
|
|
||||||
@ -0,0 +1,159 @@
|
|||||||
|
/*
|
||||||
|
* Copyright (C) 2016 Álinson Santos Xavier <isoron@gmail.com>
|
||||||
|
*
|
||||||
|
* This file is part of Loop Habit Tracker.
|
||||||
|
*
|
||||||
|
* Loop Habit Tracker is free software: you can redistribute it and/or modify
|
||||||
|
* it under the terms of the GNU General Public License as published by the
|
||||||
|
* Free Software Foundation, either version 3 of the License, or (at your
|
||||||
|
* option) any later version.
|
||||||
|
*
|
||||||
|
* Loop Habit Tracker is distributed in the hope that it will be useful, but
|
||||||
|
* WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
|
||||||
|
* or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
|
||||||
|
* more details.
|
||||||
|
*
|
||||||
|
* You should have received a copy of the GNU General Public License along
|
||||||
|
* with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||||
|
*/
|
||||||
|
|
||||||
|
package org.isoron.uhabits.io;
|
||||||
|
|
||||||
|
import org.isoron.helpers.DateHelper;
|
||||||
|
import org.isoron.uhabits.models.CheckmarkList;
|
||||||
|
import org.isoron.uhabits.models.Habit;
|
||||||
|
import org.isoron.uhabits.models.ScoreList;
|
||||||
|
|
||||||
|
import java.io.File;
|
||||||
|
import java.io.FileInputStream;
|
||||||
|
import java.io.FileOutputStream;
|
||||||
|
import java.io.FileWriter;
|
||||||
|
import java.io.IOException;
|
||||||
|
import java.text.SimpleDateFormat;
|
||||||
|
import java.util.LinkedList;
|
||||||
|
import java.util.List;
|
||||||
|
import java.util.zip.ZipEntry;
|
||||||
|
import java.util.zip.ZipOutputStream;
|
||||||
|
|
||||||
|
public class HabitsExporter
|
||||||
|
{
|
||||||
|
private List<Habit> habits;
|
||||||
|
|
||||||
|
private List<String> generateDirs;
|
||||||
|
private List<String> generateFilenames;
|
||||||
|
|
||||||
|
private String exportDirName;
|
||||||
|
|
||||||
|
public HabitsExporter(List<Habit> habits, String exportDirName)
|
||||||
|
{
|
||||||
|
this.habits = habits;
|
||||||
|
this.exportDirName = exportDirName;
|
||||||
|
|
||||||
|
if(!this.exportDirName.endsWith("/"))
|
||||||
|
this.exportDirName += "/";
|
||||||
|
|
||||||
|
generateDirs = new LinkedList<>();
|
||||||
|
generateFilenames = new LinkedList<>();
|
||||||
|
}
|
||||||
|
|
||||||
|
private void writeHabits() throws IOException
|
||||||
|
{
|
||||||
|
String filename = "habits.csv";
|
||||||
|
new File(exportDirName).mkdirs();
|
||||||
|
FileWriter out = new FileWriter(exportDirName + filename);
|
||||||
|
generateFilenames.add(filename);
|
||||||
|
Habit.writeCSV(habits, out);
|
||||||
|
out.close();
|
||||||
|
|
||||||
|
for(Habit h : habits)
|
||||||
|
{
|
||||||
|
String habitDirName = String.format("%s/", h.name);
|
||||||
|
new File(exportDirName + habitDirName).mkdirs();
|
||||||
|
generateDirs.add(habitDirName);
|
||||||
|
|
||||||
|
writeScores(habitDirName, h.scores);
|
||||||
|
writeCheckmarks(habitDirName, h.checkmarks);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private void writeScores(String habitDirName, ScoreList scores) throws IOException
|
||||||
|
{
|
||||||
|
String path = habitDirName + "scores.csv";
|
||||||
|
FileWriter out = new FileWriter(exportDirName + path);
|
||||||
|
generateFilenames.add(path);
|
||||||
|
scores.writeCSV(out);
|
||||||
|
out.close();
|
||||||
|
}
|
||||||
|
|
||||||
|
private void writeCheckmarks(String habitDirName, CheckmarkList checkmarks) throws IOException
|
||||||
|
{
|
||||||
|
String filename = habitDirName + "checkmarks.csv";
|
||||||
|
FileWriter out = new FileWriter(exportDirName + filename);
|
||||||
|
generateFilenames.add(filename);
|
||||||
|
checkmarks.writeCSV(out);
|
||||||
|
out.close();
|
||||||
|
}
|
||||||
|
|
||||||
|
private String writeZipFile() throws IOException
|
||||||
|
{
|
||||||
|
SimpleDateFormat dateFormat = DateHelper.getCSVDateFormat();
|
||||||
|
String date = dateFormat.format(DateHelper.getStartOfToday());
|
||||||
|
String zipFilename = String.format("%s/habits-%s.zip", exportDirName, date);
|
||||||
|
|
||||||
|
FileOutputStream fos = new FileOutputStream(zipFilename);
|
||||||
|
ZipOutputStream zos = new ZipOutputStream(fos);
|
||||||
|
|
||||||
|
for(String filename : generateFilenames)
|
||||||
|
addFileToZip(zos, filename);
|
||||||
|
|
||||||
|
zos.close();
|
||||||
|
fos.close();
|
||||||
|
|
||||||
|
return zipFilename;
|
||||||
|
}
|
||||||
|
|
||||||
|
private void addFileToZip(ZipOutputStream zos, String filename) throws IOException
|
||||||
|
{
|
||||||
|
FileInputStream fis = new FileInputStream(new File(exportDirName + filename));
|
||||||
|
ZipEntry ze = new ZipEntry(filename);
|
||||||
|
zos.putNextEntry(ze);
|
||||||
|
|
||||||
|
int length;
|
||||||
|
byte bytes[] = new byte[1024];
|
||||||
|
while((length = fis.read(bytes)) >= 0)
|
||||||
|
zos.write(bytes, 0, length);
|
||||||
|
|
||||||
|
zos.closeEntry();
|
||||||
|
fis.close();
|
||||||
|
}
|
||||||
|
|
||||||
|
public String writeArchive()
|
||||||
|
{
|
||||||
|
String zipFilename;
|
||||||
|
|
||||||
|
try
|
||||||
|
{
|
||||||
|
writeHabits();
|
||||||
|
zipFilename = writeZipFile();
|
||||||
|
cleanup();
|
||||||
|
}
|
||||||
|
catch (IOException e)
|
||||||
|
{
|
||||||
|
e.printStackTrace();
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
|
||||||
|
return zipFilename;
|
||||||
|
}
|
||||||
|
|
||||||
|
private void cleanup()
|
||||||
|
{
|
||||||
|
for(String filename : generateFilenames)
|
||||||
|
new File(exportDirName + filename).delete();
|
||||||
|
|
||||||
|
for(String filename : generateDirs)
|
||||||
|
new File(exportDirName + filename).delete();
|
||||||
|
|
||||||
|
new File(exportDirName).delete();
|
||||||
|
}
|
||||||
|
}
|
||||||
Loading…
Reference in new issue